| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | shard_test.py | 102 def _build_dataset(self, num_elements, num_shards, index): argument
111 def test(self, verify_fn, elems, num_shards, index): argument
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | shard_dataset_op.cc | 44 Dataset(OpKernelContext* ctx, int64_t num_shards, int64_t index, in Dataset()
103 Node* num_shards = nullptr; in AsGraphDefInternal() local
257 int64_t num_shards = 0; in MakeDataset() local
|
| D | shard_dataset_op_test.cc | 25 ShardDatasetParams(T input_dataset_params, int64_t num_shards, int64_t index, in ShardDatasetParams()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_embedding_optimization_parameters_utils.cc | 425 int num_shards; in operator ()() local
455 int num_shards; in operator ()() local
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | naming.cc | 28 string DataFilename(StringPiece prefix, int32_t shard_id, int32_t num_shards) { in DataFilename()
|
| D | tensor_bundle.cc | 579 int num_shards = 0; member
611 int num_shards; in MergeOneBundle() 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
817 def _random_weights(self, vocab_size=4, embed_dim=4, num_shards=1): 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/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 | 100 def __init__(self, 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_t 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 | 69 int64_t num_shards = 0; in RunWithFixedBlockSize() local
121 int64_t num_shards = 0; in RunWithFixedBlockSizeTransformRangeConcurrently() local
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_optimizer.py | 77 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_t num_shards, int32_t shard) in FixedUnigramSampler()
255 int32_t num_shards, int32_t shard) in FixedUnigramSampler()
|
| D | candidate_sampler_ops.cc | 171 int64_t num_shards; in FixedUnigramCandidateSamplerOp() local
|
| D | topk_op_gpu.h | 262 __device__ void mergeShards(int num_shards, int k, in mergeShards()
373 cudaError LaunchTopKKernel(const gpuStream_t& stream, int num_shards, in LaunchTopKKernel()
|
| /external/tensorflow/tensorflow/core/data/ |
| D | split_utils.cc | 67 int64_t num_shards, int64_t shard_index, in ShardingSplitProvider()
|
| /external/autotest/server/cros/ap_configurators/ |
| D | pyauto_utils.py | 170 def Shard(ilist, shard_index, num_shards): argument
|