Searched defs:shard_id (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | gmm_ops.py | 282 def _define_full_covariance_probs(self, shard_id, shard): argument
304 def _define_diag_covariance_probs(self, shard_id, shard): argument
328 def _define_log_prob_operation(self, shard_id, shard): argument
345 def _define_prior_log_prob_operation(self, shard_id): argument
357 def _define_expectation_operation(self, shard_id): argument
370 def _define_partial_maximization_operation(self, shard_id, shard): 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/lite/tools/accuracy/ilsvrc/ |
| D | imagenet_accuracy_eval.cc | 68 uint64_t shard_id, const ImagenetTopKAccuracy::AccuracyStats& stats, in OnSingleImageEvaluationComplete()
104 uint64_t shard_id, const ImagenetTopKAccuracy::AccuracyStats& stats, in OnSingleImageEvaluationComplete()
|
| D | imagenet_model_evaluator.cc | 100 uint64_t shard_id, const ImagenetTopKAccuracy::AccuracyStats& stats, in OnSingleImageEvaluationComplete()
178 Status EvaluateModelForShard(const uint64_t shard_id, in EvaluateModelForShard()
324 const uint64_t shard_id = i + 1; in EvaluateModel() local
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | tensor_bundle.proto | 49 int32 shard_id = 3; field
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | tpu_embedding_ops.cc | 154 int shard_id; in RegisterPerTableLoadOpsForAlgorithmBody() local
271 int shard_id; in RegisterPerTableRetrieveOpsForAlgorithmBody() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_ops_3d.cc | 531 for (int shard_id = start; shard_id < limit; ++shard_id) { in Compute() local
951 for (int shard_id = start; shard_id < limit; ++shard_id) { in Compute() local
|
| D | conv_grad_filter_ops.cc | 371 for (int shard_id = start; shard_id < limit; ++shard_id) { in Compute() local
|
| D | conv_grad_input_ops.cc | 508 for (int shard_id = start; shard_id < limit; ++shard_id) { in Compute() local
|