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