HomeSort by relevance Sort by last modified time
    Searched full:set_shard_dimensions (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/contrib/tpu/python/tpu/
tpu_infeed_test.py 96 i.set_shard_dimensions([1, 0])
115 i.set_shard_dimensions([0, 0])
123 i.set_shard_dimensions([1, 0])
tpu_feed.py 107 self.set_shard_dimensions(shard_dimensions)
224 def set_shard_dimensions(self, shard_dimensions): member in class:InfeedQueue

Completed in 76 milliseconds