Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/
Dnon_max_suppression.cc97 const TfLiteTensor* input_max_output_size; in Prepare() local
100 &input_max_output_size)); in Prepare()
101 TF_LITE_ENSURE_EQ(context, input_max_output_size->type, kTfLiteInt32); in Prepare()
102 TF_LITE_ENSURE_EQ(context, NumDimensions(input_max_output_size), 0); in Prepare()
103 const bool is_max_output_size_const = IsConstantTensor(input_max_output_size); in Prepare()
106 max_output_size_value = *GetTensorData<int>(input_max_output_size); in Prepare()
211 const TfLiteTensor* input_max_output_size; in Eval() local
214 &input_max_output_size)); in Eval()
215 const int max_output_size_value = *GetTensorData<int>(input_max_output_size); in Eval()
217 const bool is_max_output_size_const = IsConstantTensor(input_max_output_size); in Eval()