set_number_of_partitions (reference) in projects: external

Searched refs:set_number_of_partitions (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
D	tpu_sharding.py	`101 def set_number_of_partitions(self, number_of_partitions): member in ShardingPolicy`
D	tpu_sharding_test.py	`115 p.set_number_of_partitions(4)`
D	tpu_feed.py	`374 policy.set_number_of_partitions(self._number_of_partitions)`