set_number_of_shards (reference) in projects: external

Searched refs:set_number_of_shards (Results 1 – 7 of 7) 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(3) 124 p.set_number_of_shards(2)`
D	tpu_function.py	`37 def set_number_of_shards(self, number_of_shards): member in TpuContext 57 _current_tpu_context.set_number_of_shards(number_of_shards) 60 _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	`70 def set_number_of_shards(self, number_of_shards): member in ShardingPolicy 160 self.set_number_of_shards(other.number_of_shards)`
D	tpu_feed.py	`361 def set_number_of_shards(self, number_of_shards): member in InfeedQueue 373 policy.set_number_of_shards(number_of_shards) 422 self.set_number_of_shards(number_of_shards)`
D	training_loop.py	`125 infeed_queue.set_number_of_shards(number_of_shards)`
D	tpu.py	`1451 infeed_queue.set_number_of_shards(num_replicas)`