LostTech.TensorFlow : API Documentation

Type IShardingPolicy

Namespace tensorflow.python.tpu.tpu_sharding

Interfaces IPythonObjectContainer

Public instance methods

object get_sharded_shape(object shape, object shard_index)

object get_unsharded_shape(object shapes)

object set_number_of_shards_(object number_of_shards)

object set_shard_dimension_(object shard_dimension)

Public properties

object number_of_shards get;

object shard_dimension get;