Searched refs:logits_size (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | multinomial.cc | 37 const FloatType* logits, int logits_size, in MultinomialSample() argument
42 cumulative_odds.reserve(logits_size); in MultinomialSample()
47 for (int i = 0; i < logits_size; i++) { in MultinomialSample()
51 for (int i = 0; i < logits_size; i++) { in MultinomialSample()
76 const FloatType* logits, int logits_size, in MultinomialSample() argument
82 rng, logits, logits_size, in MultinomialSample()
87 rng, logits, logits_size, in MultinomialSample()
101 int logits_size, TfLiteTensor* output, in MultinomialSample() argument
113 logits_size, output, outputs_offset, output_size)); in MultinomialSample()
120 logits_size, output, outputs_offset, output_size)); in MultinomialSample()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 2487 const int64 logits_size = CalculateTensorElementCount( in PredictSoftmax() local
2504 logits_size + in PredictSoftmax()
|