| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | non_max_suppression.h | 91 const float soft_nms_sigma, int* selected_indices, in NonMaxSuppression() argument
120 if (soft_nms_sigma > 0.0) { in NonMaxSuppression()
121 scale = -0.5 / soft_nms_sigma; in NonMaxSuppression()
150 if (soft_nms_sigma > 0.0) { in NonMaxSuppression()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_NonMaxSuppressionV5.pbtxt | 38 name: "soft_nms_sigma"
41 al (c.f. https://arxiv.org/abs/1704.04503). When `soft_nms_sigma=0.0` (which
65 `soft_nms_sigma>0`)
97 To enable this Soft-NMS mode, set the `soft_nms_sigma` parameter to be
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | non_max_suppression.cc | 237 const float soft_nms_sigma = *GetTensorData<float>(input_sigma); in Eval() local
238 if (soft_nms_sigma < 0) { in Eval()
240 soft_nms_sigma); in Eval()
261 max_output_size_value, iou_threshold, score_threshold, soft_nms_sigma, in Eval()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | non_max_suppression_test.cc | 274 const float soft_nms_sigma = 0.5; in TEST() local
284 score_threshold, soft_nms_sigma, selected_indices.data(), in TEST()
298 score_threshold, soft_nms_sigma, selected_indices.data(), in TEST()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | NonMaxSuppressionV5.pbtxt | 24 name: "soft_nms_sigma"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | NonMaxSuppressionV5.pbtxt | 24 name: "soft_nms_sigma"
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cc | 165 const T score_threshold, const T soft_nms_sigma, in DoNonMaxSuppressionOp() argument
195 bool is_soft_nms = soft_nms_sigma > static_cast<T>(0.0); in DoNonMaxSuppressionOp()
197 scale = static_cast<T>(-0.5) / soft_nms_sigma; in DoNonMaxSuppressionOp()
821 const Tensor& soft_nms_sigma = context->input(5); in Compute() local
823 context, TensorShapeUtils::IsScalar(soft_nms_sigma.shape()), in Compute()
825 soft_nms_sigma.shape().DebugString())); in Compute()
826 const T soft_nms_sigma_val = soft_nms_sigma.scalar<T>()(); in Compute()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_patterns.td | 411 $score_threshold, $soft_nms_sigma, $pad_to_max_output_size),
413 $score_threshold, $soft_nms_sigma)>;
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | image_ops.cc | 180 ShapeHandle soft_nms_sigma; in SoftNMSShapeFn() local
181 TF_RETURN_IF_ERROR(c->WithRank(c->input(5), 0, &soft_nms_sigma)); in SoftNMSShapeFn()
|
| D | ops.pbtxt | 26597 name: "soft_nms_sigma"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.image.pbtxt | 141 …es\', \'max_output_size\', \'iou_threshold\', \'score_threshold\', \'soft_nms_sigma\', \'name\'], …
|
| D | tensorflow.raw_ops.pbtxt | 2665 …es\', \'max_output_size\', \'iou_threshold\', \'score_threshold\', \'soft_nms_sigma\', \'pad_to_ma…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.image.pbtxt | 145 …es\', \'max_output_size\', \'iou_threshold\', \'score_threshold\', \'soft_nms_sigma\', \'name\'], …
|
| D | tensorflow.raw_ops.pbtxt | 2665 …es\', \'max_output_size\', \'iou_threshold\', \'score_threshold\', \'soft_nms_sigma\', \'pad_to_ma…
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_impl.py | 3704 soft_nms_sigma=0.0, argument
3770 soft_nms_sigma = ops.convert_to_tensor(
3771 soft_nms_sigma, name='soft_nms_sigma')
3779 soft_nms_sigma,
|
| D | image_ops_test.py | 5014 soft_nms_sigma = constant_op.constant(soft_nms_sigma_np, dtype=dtype)
5017 soft_nms_sigma)
5054 soft_nms_sigma = constant_op.constant(soft_nms_sigma_np)
5062 soft_nms_sigma)
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.td | 1353 To enable this Soft-NMS mode, set the `soft_nms_sigma` parameter to be
1363 TFL_FpTensor:$soft_nms_sigma
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 8989 To enable this Soft-NMS mode, set the `soft_nms_sigma` parameter to be
9004 al (c.f. https://arxiv.org/abs/1704.04503). When `soft_nms_sigma=0.0` (which
9005 is default), we fall back to standard (hard) NMS.}]>:$soft_nms_sigma,
9016 `soft_nms_sigma>0`)}]>:$selected_scores,
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 11845 …_size tf.Output, iou_threshold tf.Output, score_threshold tf.Output, soft_nms_sigma tf.Output, opt…
11856 boxes, scores, max_output_size, iou_threshold, score_threshold, soft_nms_sigma,
|