| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | shard_dataset_serialization_test.py | 31 def _build_dataset(self, num_elements, num_shards, index): argument
35 def testCore(self, elems, num_shards, index1, index2): argument
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | embedding_ops_test.py | 47 def _random_weights(self, vocab_size=4, embed_dim=4, num_shards=1): argument
275 def _random_weights(self, size=50, num_shards=1): argument
408 def _random_weights(self, size=50, num_shards=1): argument
583 def _EmbeddingParams(num_shards, argument
615 num_shards, argument
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | shard_dataset_op.cc | 35 int64 num_shards = 0; in MakeDataset() local
57 Dataset(OpKernelContext* ctx, int64 num_shards, int64 index, in Dataset()
101 Node* num_shards = nullptr; in AsGraphDefInternal() local
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | factorization_ops.py | 329 def _shard_sizes(cls, dims, num_shards): argument
335 def _create_factors(cls, rows, cols, num_shards, init, name): argument
369 def _create_weights(cls, wt_init, num_wts, num_shards, name): argument
633 def _get_sharding_func(size, num_shards): argument
|
| /external/tensorflow/tensorflow/core/util/tensor_bundle/ |
| D | naming.cc | 28 string DataFilename(StringPiece prefix, int32 shard_id, int32 num_shards) { in DataFilename()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_ops.py | 36 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 | 384 num_shards=1): argument
421 num_shards, gpu_indices): argument
|
| /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/contrib/boosted_trees/lib/utils/ |
| D | parallel_for.cc | 30 const int num_shards = std::max<int>( in ParallelFor() local
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | tensor_bundle.proto | 25 int32 num_shards = 1; field
|
| /external/tensorflow/tensorflow/core/util/ |
| D | work_sharder.cc | 65 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
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | tpu_embedding_ops.cc | 152 int num_shards; in RegisterPerTableLoadOpsForAlgorithmBody() local
269 int num_shards; in RegisterPerTableRetrieveOpsForAlgorithmBody() local
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_optimizer.py | 61 def _verify_and_get_subgroup_size(self, group_assignment, num_shards): argument
|
| D | tpu.py | 1087 num_shards=1, argument
1237 num_shards=1, argument
1317 num_shards=1, argument
|
| D | tpu_config.py | 118 num_shards=None, 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/tensorflow/tensorflow/contrib/linear_optimizer/python/ops/ |
| D | sharded_mutable_dense_hashtable.py | 64 num_shards=1, argument
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | range_sampler.cc | 236 int32 num_shards, int32 shard) in FixedUnigramSampler()
252 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
|
| /external/tensorflow/tensorflow/contrib/linear_optimizer/python/kernel_tests/ |
| D | sdca_ops_test.py | 142 def make_variable_dict(max_age, max_gender, num_shards=None, partitioned=False): argument
|