Searched defs:set_number_of_shards (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ | ||
| D | tpu_function.py | 37 def set_number_of_shards(self, number_of_shards): member in TpuContext |
| D | tpu_sharding.py | 70 def set_number_of_shards(self, number_of_shards): member in ShardingPolicy |
| D | tpu_feed.py | 361 def set_number_of_shards(self, number_of_shards): member in InfeedQueue |