Searched refs:output_indexes (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | topk_v2.cc | 64 TfLiteTensor* output_indexes; in ResizeOutput() local
66 context, GetOutputSafe(context, node, kOutputIndexes, &output_indexes)); in ResizeOutput()
71 output_indexes->type = kTfLiteInt32; in ResizeOutput()
83 TF_LITE_ENSURE_OK(context, resize_tensor(output_indexes, output_indexes_shape, in ResizeOutput()
177 int32* output_indexes, T* output_values) { in TopK() argument
187 int32* indexes_row = output_indexes + row * k; in TopK()
219 TfLiteTensor* output_indexes; in Prepare() local
221 context, GetOutputSafe(context, node, kOutputIndexes, &output_indexes)); in Prepare()
225 SetTensorToDynamic(output_indexes); in Prepare()
235 TfLiteTensor* output_indexes; in Eval() local
[all …]
|
| D | unique.cc | 76 TfLiteTensor* output_indexes; in EvalImpl() local
77 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 1, &output_indexes)); in EvalImpl()
79 I* indexes = GetTensorData<I>(output_indexes); in EvalImpl()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 1469 auto& output_indexes = model->GetArray(op->outputs[1]); in ProcessTopkV2Operator() local
1472 if (output_indexes.has_shape()) { in ProcessTopkV2Operator()
1486 auto output_indexes_dims = output_indexes.mutable_shape()->mutable_dims(); in ProcessTopkV2Operator()
|