| /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/kernels/data/ |
| D | shard_dataset_op.cc | 43 Dataset(OpKernelContext* ctx, int64 num_shards, int64 index, in Dataset()
102 Node* num_shards = nullptr; in AsGraphDefInternal() local
247 int64 num_shards = 0; in MakeDataset() local
|
| D | split_utils.cc | 61 int64 num_shards, int64 shard_index, in ShardingSplitProvider()
|
| D | shard_dataset_op_test.cc | 25 ShardDatasetParams(T input_dataset_params, int64 num_shards, int64 index, in ShardDatasetParams()
|
| /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 | 577 int num_shards = 0; member
609 int num_shards; in MergeOneBundle() local
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_embedding_optimization_parameters_utils.cc | 367 int num_shards; in operator ()() local
397 int num_shards; in operator ()() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | embedding_ops_test.py | 140 def _EmbeddingParams(num_shards, argument
170 def _EmbeddingParamsAsPartitionedVariable(num_shards, argument
190 num_shards, argument
797 def _random_weights(self, vocab_size=4, embed_dim=4, num_shards=1): argument
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_ops.py | 30 def auto_shard_dataset(dataset, num_shards, index, num_replicas_in_sync=None): argument
|
| D | sharded_variable.py | 97 def __init__(self, num_shards): argument
|
| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | all_reduce_test.py | 189 def _buildShuffle(self, num_workers, num_gpus, num_shards): argument
196 def _testShuffleAllReduce(self, num_workers, num_gpus, shape, num_shards): argument
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | tensor_bundle.proto | 27 int32 num_shards = 1; field
|
| /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/python/tpu/ |
| D | tpu_optimizer.py | 72 def _verify_and_get_subgroup_size(self, group_assignment, num_shards): argument
|
| D | tpu_feed.py | 726 def split_fn(inp, num_shards, axis, name): 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 | 229 num_shards=1, argument
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | functional_saver.py | 120 def sharded_filename(filename_tensor, shard, 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 | 170 def Shard(ilist, shard_index, num_shards): argument
|
| /external/tensorflow/tensorflow/python/training/ |
| D | saver.py | 182 def sharded_filename(self, filename_tensor, shard, num_shards): argument
|