Home
last modified time | relevance | path

Searched refs:output_indexes_shape (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dtopk_v2.cc49 TfLiteIntArray* output_indexes_shape = TfLiteIntArrayCreate(num_dimensions); in ResizeOutput() local
52 output_indexes_shape->data[i] = input->dims->data[i]; in ResizeOutput()
55 output_indexes_shape->data[num_dimensions - 1] = k; in ResizeOutput()
72 TF_LITE_ENSURE_OK(context, resize_tensor(output_indexes, output_indexes_shape, in ResizeOutput()