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,
|