Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dtopk_v2.cc27 constexpr int kInputTopK = 1; variable
33 const TfLiteTensor* top_k = GetInput(context, node, kInputTopK); in ResizeOutput()
170 const TfLiteTensor* top_k = GetInput(context, node, kInputTopK); in Prepare()
191 const TfLiteTensor* top_k = GetInput(context, node, kInputTopK); in Eval()