Searched defs:number_of_shards (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/tpu/python/tpu/
H A Dtpu_function.py35 def number_of_shards(self): member in class:TpuContext
38 def set_number_of_shards(self, number_of_shards):
39 self._number_of_shards = number_of_shards
48 def tpu_shard_context(number_of_shards):
49 if _current_tpu_context.number_of_shards is not None:
52 _current_tpu_context.set_number_of_shards(number_of_shards)
H A Dtpu_sharding.py41 if self.number_of_shards is None or self.shard_dimension is None:
45 (self.number_of_shards, self.shard_dimension))
65 def number_of_shards(self): member in class:ShardingPolicy
69 def set_number_of_shards(self, number_of_shards):
72 If the policy has been frozen then number_of_shards must match the
76 number_of_shards: The number of shards to use in the policy.
79 ValueError: If the policy has been frozen and number_of_shards
80 differs from the frozen value; or number_of_shards <= 0.
83 if self._number_of_shards != number_of_shards:
86 "frozen to use %d." % (number_of_shards, sel
[all...]
H A Dtpu_feed.py250 def number_of_shards(self): member in class:InfeedQueue
257 return self._sharding_policies[0].number_of_shards
259 def set_number_of_shards(self, number_of_shards):
263 number_of_shards: number of ways to shard the InfeedQueue.
266 ValueError: if number_of_shards is not > 0; or the policies have
267 been frozen and number_of_shards was already set to something
271 policy.set_number_of_shards(number_of_shards)
319 number_of_shards = len(input_tensors)
320 self.set_number_of_shards(number_of_shards)
337 for i in xrange(1, self.number_of_shards)
[all...]

Completed in 178 milliseconds