Searched refs:MaxSizePartitioner (Results 1 – 4 of 4) sorted by relevance
1 path: "tensorflow.distribute.experimental.partitioners.MaxSizePartitioner"3 is_instance: "<class \'tensorflow.python.distribute.sharded_variable.MaxSizePartitioner\'>"
8 name: "MaxSizePartitioner"
88 partitioner = sharded_variable.MaxSizePartitioner(max_shard_bytes=4)92 partitioner = sharded_variable.MaxSizePartitioner(97 partitioner = sharded_variable.MaxSizePartitioner(max_shard_bytes=1024)
168 class MaxSizePartitioner(Partitioner): class