Searched defs:set_number_of_shards (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ | ||
| H A D | tpu_function.py | 33 def set_number_of_shards(self, number_of_shards): member in TpuContext |
| H A D | tpu_sharding.py | 63 def set_number_of_shards(self, number_of_shards): member in ShardingPolicy |
| H A D | tpu_feed.py | 360 def set_number_of_shards(self, number_of_shards): member in InfeedQueue |