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