| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | shard_dataset_serialization_test.py | 33 def _build_dataset(self, num_elements, num_shards, index): argument
41 def testCore(self, elems, num_shards, index): argument
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | naming.cc | 28 string DataFilename(StringPiece prefix, int32 shard_id, int32 num_shards) { in DataFilename()
|
| D | tensor_bundle.cc | 560 int num_shards = 0; member
592 int num_shards; in MergeOneBundle() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | shard_dataset_op.cc | 42 Dataset(OpKernelContext* ctx, int64 num_shards, int64 index, in Dataset()
92 Node* num_shards = nullptr; in AsGraphDefInternal() local
217 int64 num_shards = 0; in MakeDataset() local
|
| D | shard_dataset_op_test.cc | 25 ShardDatasetParams(T input_dataset_params, int64 num_shards, int64 index, in ShardDatasetParams()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_ops.py | 30 def auto_shard_dataset(dataset, num_shards, index): argument
|
| D | all_reduce_test.py | 190 def _buildShuffle(self, num_workers, num_gpus, num_shards): argument
197 def _testShuffleAllReduce(self, num_workers, num_gpus, shape, num_shards): argument
|
| D | cross_device_utils.py | 425 num_shards=1): argument
462 num_shards, gpu_indices): argument
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | tensor_bundle.proto | 25 int32 num_shards = 1; field
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | embedding_ops_test.py | 139 def _EmbeddingParams(num_shards, argument
169 def _EmbeddingParamsAsPartitionedVariable(num_shards, argument
189 num_shards, argument
792 def _random_weights(self, vocab_size=4, embed_dim=4, num_shards=1): argument
|
| /external/tensorflow/tensorflow/core/util/ |
| D | work_sharder.cc | 67 const int num_shards = in Do() local
|
| D | work_sharder_test.cc | 34 int64 num_shards = 0; in RunSharding() local
|
| /external/tensorflow/tensorflow/core/lib/core/ |
| D | blocking_counter_test.cc | 57 const int num_shards = num_threads * shards_per_thread; in BM_BlockingCounter() local
|
| D | threadpool_test.cc | 68 int64 num_shards = 0; in RunWithFixedBlockSize() local
120 int64 num_shards = 0; in RunWithFixedBlockSizeTransformRangeConcurrently() local
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_embedding_optimization_parameters_utils.cc | 395 int num_shards; in RegisterPerTableLoadOpsForAlgorithmBody() local
518 int num_shards; in RegisterPerTableRetrieveOpsForAlgorithmBody() local
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_optimizer.py | 72 def _verify_and_get_subgroup_size(self, group_assignment, num_shards): argument
|
| D | tpu.py | 1393 num_shards=1, argument
1544 num_shards=1, argument
1625 num_shards=1, argument
|
| D | tpu_feed.py | 706 def split_fn(inp, num_shards, axis, name): argument
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | functional_saver.py | 107 def sharded_filename(filename_tensor, shard, num_shards): argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | partitioned_variables.py | 222 def fixed_size_partitioner(num_shards, axis=0): argument
|
| D | candidate_sampling_ops.py | 224 num_shards=1, argument
|
| /external/libtextclassifier/native/utils/grammar/utils/ |
| D | rules.h | 63 explicit Rules(const int num_shards = 1) : num_shards_(num_shards) {} in num_shards_() argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | range_sampler.cc | 239 int32 num_shards, int32 shard) in FixedUnigramSampler()
255 int32 num_shards, int32 shard) in FixedUnigramSampler()
|
| D | candidate_sampler_ops.cc | 171 int64 num_shards; in FixedUnigramCandidateSamplerOp() local
|
| /external/autotest/server/cros/ap_configurators/ |
| D | pyauto_utils.py | 168 def Shard(ilist, shard_index, num_shards): argument
|