Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dtopk_v2.cc62 auto resize_tensor = [context](TfLiteTensor* tensor, TfLiteIntArray* new_size, in ResizeOutput() local
72 TF_LITE_ENSURE_OK(context, resize_tensor(output_indexes, output_indexes_shape, in ResizeOutput()
75 resize_tensor(output_values, output_values_shape, nullptr)); in ResizeOutput()