Home
last modified time | relevance | path

Searched defs:output_indexes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dtopk_v2.cc64 TfLiteTensor* output_indexes; in ResizeOutput() local
177 int32* output_indexes, T* output_values) { in TopK()
219 TfLiteTensor* output_indexes; in Prepare() local
235 TfLiteTensor* output_indexes; in Eval() local
Dunique.cc76 TfLiteTensor* output_indexes; in EvalImpl() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc1469 auto& output_indexes = model->GetArray(op->outputs[1]); in ProcessTopkV2Operator() local