Home
last modified time | relevance | path

Searched defs:soft_nms_sigma (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dnon_max_suppression.h91 const float soft_nms_sigma, int* selected_indices, in NonMaxSuppression()
/external/tensorflow/tensorflow/lite/kernels/
Dnon_max_suppression.cc237 const float soft_nms_sigma = *GetTensorData<float>(input_sigma); in Eval() local
/external/tensorflow/tensorflow/lite/kernels/internal/
Dnon_max_suppression_test.cc274 const float soft_nms_sigma = 0.5; in TEST() local
/external/tensorflow/tensorflow/core/kernels/image/
Dnon_max_suppression_op.cc165 const T score_threshold, const T soft_nms_sigma, in DoNonMaxSuppressionOp()
821 const Tensor& soft_nms_sigma = context->input(5); in Compute() local
/external/tensorflow/tensorflow/core/ops/
Dimage_ops.cc180 ShapeHandle soft_nms_sigma; in SoftNMSShapeFn() local
/external/tensorflow/tensorflow/python/ops/
Dimage_ops_impl.py3704 soft_nms_sigma=0.0, argument