Home
last modified time | relevance | path

Searched refs:_num_shards (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dinput_ops_test.py41 self._num_shards = 2
96 for f in range(self._shard_index, self._num_files, self._num_shards):
106 dataset, self._num_shards, self._shard_index)
116 dataset, self._num_shards, self._shard_index)
127 dataset, self._num_shards, self._shard_index)
140 dataset, self._num_shards, self._shard_index)
146 for f in range(self._shard_index, self._num_files, self._num_shards):
172 dataset, self._num_shards, self._shard_index)
180 self._num_shards * batch_size)
187 for f in range(0, self._num_files, self._num_shards):
[all …]
/external/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/
Dsharded_mutable_dense_hashtable.py93 def _num_shards(self): member in ShardedMutableDenseHashTable
103 self._table_shards[i].size() for i in range(self._num_shards)
114 indices = math_ops.mod(math_ops.abs(keys), self._num_shards)
131 num_shards = self._num_shards
157 num_shards = self._num_shards
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py2747 self._num_shards = ops.convert_to_tensor(
2752 num_shards=self._num_shards,