Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dtopk_v2.cc50 TfLiteIntArray* output_values_shape = TfLiteIntArrayCreate(num_dimensions); in ResizeOutput() local
53 output_values_shape->data[i] = input->dims->data[i]; in ResizeOutput()
56 output_values_shape->data[num_dimensions - 1] = k; in ResizeOutput()
73 output_values_shape)); in ResizeOutput()
75 resize_tensor(output_values, output_values_shape, nullptr)); in ResizeOutput()