| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| D | topk_v2.cc | 33 TfLiteTensor* top_k = GetInput(context, node, kInputTopK); in ResizeOutput() local
35 TF_LITE_ENSURE_EQ(context, top_k->type, kTfLiteInt32); in ResizeOutput()
37 TF_LITE_ENSURE_EQ(context, NumDimensions(top_k), 1); in ResizeOutput()
38 TF_LITE_ENSURE_EQ(context, NumElements(top_k), 1); in ResizeOutput()
39 const int32 k = top_k->data.i32[0]; in ResizeOutput()
151 const auto& top_k = topc.sorted_result(); in TopK() local
152 std::copy(top_k.begin(), top_k.end(), indexes_row); in TopK()
153 std::transform(top_k.begin(), top_k.end(), output_row, in TopK()
169 TfLiteTensor* top_k = GetInput(context, node, kInputTopK); in Prepare() local
170 TF_LITE_ENSURE_EQ(context, top_k->type, kTfLiteInt32); in Prepare()
[all …]
|
| D | topk_v2_test.cc | 31 int top_k) { in TopKV2OpModel() argument
38 PopulateTensor<int32_t>(top_k_, {top_k}); in TopKV2OpModel()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | topk_op_test.py | 50 values_op, indices_op = nn_ops.top_k(inputs, k, sorted=sorted)
188 values, _ = nn_ops.top_k(inputs, k)
196 nn_ops.top_k(inputs, 4)
201 values, _ = nn_ops.top_k(inputs, 3)
226 op = nn_ops.top_k(v, k)
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_beam_search.h | 250 const bool top_k = in Step() local
253 const int max_classes = top_k ? label_selection_size_ : (num_classes_ - 1); in Step()
256 if (top_k) { in Step()
327 const int label = top_k ? top_k_indices[ind] : ind; in Step()
328 const float logit = top_k ? top_k_logits[ind] : raw_input(ind); in Step()
|
| /external/tensorflow/tensorflow/contrib/framework/python/ops/ |
| D | sort_ops.py | 79 return nn_ops.top_k(values, k)[0]
98 values, unused_indices = nn_ops.top_k(top_k_input, k)
|
| /external/tensorflow/tensorflow/examples/speech_commands/ |
| D | label_wav.py | 71 top_k = predictions.argsort()[-num_top_predictions:][::-1]
72 for node_id in top_k:
|
| D | label_wav_dir.py | 80 top_k = predictions.argsort()[-num_top_predictions:][::-1]
81 for node_id in top_k:
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | topk_op.cc | 189 std::unique_ptr<std::vector<int32>> top_k(filter.Extract()); in Compute() local
190 for (auto top_k_it = top_k->begin(); top_k_it != top_k->end(); in Compute()
|
| /external/tensorflow/tensorflow/examples/label_image/ |
| D | label_image.py | 137 top_k = results.argsort()[-5:][::-1] variable
139 for i in top_k:
|
| /external/tensorflow/tensorflow/examples/tutorials/word2vec/ |
| D | word2vec_basic.py | 291 top_k = 8 # number of nearest neighbors variable
292 nearest = (-sim[i, :]).argsort()[1:top_k + 1]
294 for k in xrange(top_k):
|
| /external/tensorflow/tensorflow/contrib/sparsemax/python/ops/ |
| D | sparsemax.py | 55 z_sorted, _ = nn.top_k(z, k=dims)
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/ |
| D | permute.py | 106 p, _ = nn_ops.top_k(-permutation,
|
| /external/tensorflow/tensorflow/contrib/image/python/ops/ |
| D | image_ops.py | 409 top_k=-1, argument
441 distance_mat, num_valid_rows, top_k, name=name)
|
| /external/tensorflow/tensorflow/contrib/nn/python/ops/ |
| D | sampling_ops.py | 102 _, resampled_indices = nn.top_k(sampled_logits, k=num_resampled, sorted=False)
|
| /external/tensorflow/tensorflow/contrib/gan/python/eval/python/ |
| D | sliced_wasserstein_impl.py | 142 sorted_tmatrix = nn_ops.top_k(tmatrix, num_rows)[0]
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | sample_stats.py | 521 sorted_, _ = nn_ops.top_k(tensor, k=array_ops.shape(tensor)[-1])
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | metrics_impl.py | 2414 _, top_k_idx = nn.top_k(predictions, k)
3115 _, predictions_idx = nn.top_k(predictions, k)
3410 _, top_k_idx = nn.top_k(predictions, k)
|
| D | nn_ops.py | 2307 def top_k(input, k=1, sorted=True, name=None): # pylint: disable=redefined-builtin function
|
| /external/pdfium/third_party/libopenjpeg20/ |
| D | jp2.c | 1042 OPJ_INT32 k, top_k; in opj_jp2_apply_pclr() local
1106 top_k = color->jp2_pclr->nr_entries - 1; in opj_jp2_apply_pclr()
1138 } else if (k > top_k) { in opj_jp2_apply_pclr()
1139 k = top_k; in opj_jp2_apply_pclr()
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| D | input_pipeline.py | 762 _, indices_descending = nn.top_k(
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| D | beam_search_decoder.py | 579 next_beam_scores, word_indices = nn_ops.top_k(scores_flat, k=next_beam_size)
|
| /external/tensorflow/tensorflow/contrib/metrics/python/ops/ |
| D | metric_ops.py | 1082 ordered_predictions, indices = nn.top_k(predictions, k=size)
1245 ordered_predictions, indices = nn.top_k(predictions, k=size, sorted=True)
|
| /external/tensorflow/tensorflow/tools/api/golden/ |
| D | tensorflow.nn.pbtxt | 328 name: "top_k"
|
| /external/tensorflow/tensorflow/docs_src/api_guides/python/ |
| D | nn.md | 263 * @{tf.nn.top_k}
|
| /external/tensorflow/tensorflow/contrib/lite/testing/ |
| D | generate_examples.py | 1755 out = tf.nn.top_k(input_value, k)
|