| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | embedding_ops_test.py | 45 def _random_weights(self, vocab_size=4, embed_dim=4, num_shards=1): argument
264 def _random_weights(self, size=50, num_shards=1): argument
396 def _random_weights(self, size=50, num_shards=1): argument
570 def _EmbeddingParams(num_shards, argument
602 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()
|
| D | tensor_bundle.cc | 531 int num_shards = 0; member
563 int num_shards; in MergeOneBundle() local
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | factorization_ops.py | 304 def _shard_sizes(cls, dims, num_shards): argument
310 def _create_factors(cls, rows, cols, num_shards, init, name): argument
344 def _create_weights(cls, wt_init, num_wts, num_shards, name): argument
608 def _get_sharding_func(size, num_shards): argument
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | tensor_bundle.proto | 25 int32 num_shards = 1; field
|
| /external/tensorflow/tensorflow/core/util/ |
| D | work_sharder.cc | 46 const int num_shards = in Shard() local
|
| D | work_sharder_test.cc | 33 int64 num_shards = 0; in RunSharding() local
|
| /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/contrib/all_reduce/python/ |
| D | all_reduce_test.py | 180 def _buildShuffle(self, num_workers, num_gpus, num_shards): argument
187 def _testShuffleAllReduce(self, num_workers, num_gpus, shape, num_shards): argument
|
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| D | tpu.py | 393 num_shards=1, argument
542 num_shards=1, argument
|
| D | tpu_config.py | 91 num_shards=None, argument
|
| D | tpu_feed.py | 595 def split_fn(inp, num_shards, axis, name): argument
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | embedding_ops_test.py | 131 def _EmbeddingParams(num_shards, argument
161 def _EmbeddingParamsAsPartitionedVariable(num_shards, argument
181 num_shards, argument
|
| /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/contrib/linear_optimizer/python/ops/ |
| D | sharded_mutable_dense_hashtable.py | 51 num_shards=1, argument
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | partitioned_variables.py | 221 def fixed_size_partitioner(num_shards, axis=0): argument
|
| D | candidate_sampling_ops.py | 210 num_shards=1, argument
|
| D | nn_test.py | 522 def _ShardTestEmbeddings(self, weights, biases, num_shards): 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
|
| D | topk_op_gpu.cu.cc | 269 __device__ void mergeShards(int num_shards, int k, in mergeShards()
370 cudaError LaunchTopKKernel(const cudaStream_t& stream, int num_shards, in LaunchTopKKernel()
|
| /external/autotest/server/cros/ap_configurators/ |
| D | pyauto_utils.py | 168 def Shard(ilist, shard_index, num_shards): argument
|
| /external/tensorflow/tensorflow/contrib/data/python/ops/ |
| D | dataset_ops.py | 358 def shard(self, num_shards, index): argument
|
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| D | rnn_cell.py | 45 def _get_concat_variable(name, shape, dtype, num_shards): argument
62 def _get_sharded_variable(name, shape, dtype, num_shards): argument
|
| /external/tensorflow/tensorflow/python/training/ |
| D | saver.py | 302 def sharded_filename(self, filename_tensor, shard, num_shards): argument
|