Searched refs:input_sigma (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | non_max_suppression.cc | 125 const TfLiteTensor* input_sigma; in Prepare() local
127 context, GetInputSafe(context, node, kInputTensorSigma, &input_sigma)); in Prepare()
128 TF_LITE_ENSURE_EQ(context, input_sigma->type, kTfLiteFloat32); in Prepare()
129 TF_LITE_ENSURE_EQ(context, NumDimensions(input_sigma), 0); in Prepare()
234 const TfLiteTensor* input_sigma; in Eval() local
236 context, GetInputSafe(context, node, kInputTensorSigma, &input_sigma)); in Eval()
237 const float soft_nms_sigma = *GetTensorData<float>(input_sigma); in Eval()
|