set_number_of_shards (reference) in projects: external

Searched refs:set_number_of_shards (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
D	tpu_sharding_test.py	`38 p2.set_number_of_shards(17) 49 p1.set_number_of_shards(17) 57 p1.set_number_of_shards(17) 65 p1.set_number_of_shards(17) 80 p1.set_number_of_shards(1) 84 p1.set_number_of_shards(17) 93 p.set_number_of_shards(3) 113 p.set_number_of_shards(2)`
D	tpu_function.py	`36 def set_number_of_shards(self, number_of_shards): member in TpuContext 50 _current_tpu_context.set_number_of_shards(number_of_shards) 53 _current_tpu_context.set_number_of_shards(None)`
D	tpu_infeed_test.py	`83 i.set_number_of_shards(2) 85 i.set_number_of_shards(3) 97 i.set_number_of_shards(3) 99 i.set_number_of_shards(4) 112 i.set_number_of_shards(2) 117 i.set_number_of_shards(1)`
D	tpu_sharding.py	`69 def set_number_of_shards(self, number_of_shards): member in ShardingPolicy 133 self.set_number_of_shards(other.number_of_shards)`
D	tpu_feed.py	`344 def set_number_of_shards(self, number_of_shards): member in InfeedQueue 356 policy.set_number_of_shards(number_of_shards) 404 self.set_number_of_shards(number_of_shards)`
D	training_loop.py	`117 infeed_queue.set_number_of_shards(number_of_shards)`
D	tpu.py	`869 infeed_queue.set_number_of_shards(num_replicas)`
D	tpu_estimator.py	`832 infeed_queue.set_number_of_shards(num_of_replicas_per_host)`