LostTech.TensorFlow : API Documentation

Type ShardingPolicy

Namespace tensorflow.python.tpu.tpu_sharding

Parent PythonObjectContainer

Interfaces IShardingPolicy

Public instance methods

object get_sharded_shape(IEnumerable<int> shape, Nullable<int> shard_index)

object get_sharded_shape(TensorShape shape, Nullable<int> shard_index)

object get_sharded_shape(string shape, Nullable<int> shard_index)

object get_sharded_shape_dyn(object shape, object shard_index)

TensorShape get_unsharded_shape(IEnumerable<object> shapes)

object get_unsharded_shape_dyn(object shapes)

void merge(ShardingPolicy other)

void set_shard_dimension_(Nullable<int> shard_dimension)

object set_shard_dimension__dyn(object shard_dimension)

Public properties

Nullable<int> number_of_shards get;

object number_of_shards_dyn get;

object PythonObject get;

Dimension shard_dimension get;

object shard_dimension_dyn get;