Searched refs:output_indexes (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | topk_v2.cc | 57 TfLiteTensor* output_indexes = GetOutput(context, node, kOutputIndexes); in ResizeOutput() local
60 output_indexes->type = kTfLiteInt32; in ResizeOutput()
72 TF_LITE_ENSURE_OK(context, resize_tensor(output_indexes, output_indexes_shape, in ResizeOutput()
139 int32* output_indexes, T* output_values) { in TopK() argument
149 int32* indexes_row = output_indexes + row * k; in TopK()
177 TfLiteTensor* output_indexes = GetOutput(context, node, kOutputIndexes); in Prepare() local
179 SetTensorToDynamic(output_indexes); in Prepare()
187 TfLiteTensor* output_indexes = GetOutput(context, node, kOutputIndexes); in Eval() local
203 TopK(row_size, num_rows, input->data.f, k, output_indexes->data.i32, in Eval()
207 TopK(row_size, num_rows, input->data.uint8, k, output_indexes->data.i32, in Eval()
[all …]
|
| D | unique.cc | 67 TfLiteTensor* output_indexes = GetOutput(context, node, 1); in EvalImpl() local
68 I* indexes = GetTensorData<I>(output_indexes); in EvalImpl()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 1449 auto& output_indexes = model->GetArray(op->outputs[1]); in ProcessTopkV2Operator() local
1452 if (output_indexes.has_shape()) { in ProcessTopkV2Operator()
1466 auto output_indexes_dims = output_indexes.mutable_shape()->mutable_dims(); in ProcessTopkV2Operator()
|