Searched refs:_num_shards (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_ops_test.py | 41 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/ |
| D | sharded_mutable_dense_hashtable.py | 93 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/ |
| D | dataset_ops.py | 2747 self._num_shards = ops.convert_to_tensor(
2752 num_shards=self._num_shards,
|