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.cc57 TfLiteTensor* output_indexes = GetOutput(context, node, kOutputIndexes); in ResizeOutput() local
139 int32* output_indexes, T* output_values) { in TopK()
177 TfLiteTensor* output_indexes = GetOutput(context, node, kOutputIndexes); in Prepare() local
187 TfLiteTensor* output_indexes = GetOutput(context, node, kOutputIndexes); in Eval() local
Dunique.cc67 TfLiteTensor* output_indexes = GetOutput(context, node, 1); in EvalImpl() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc1449 auto& output_indexes = model->GetArray(op->outputs[1]); in ProcessTopkV2Operator() local