| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | non_max_suppression_test.cc | 70 const float iou_threshold = 0.5; in TEST() local
97 const float iou_threshold = 0.5; in TEST() local
126 const float iou_threshold = 0.5; in TEST() local
168 const float iou_threshold = 0.5; in TEST() local
205 const float iou_threshold = 0.5; in TEST() local
272 const float iou_threshold = 1.0; in TEST() local
309 const float iou_threshold = 0.5; in TEST() local
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cu.cc | 73 const T iou_threshold) { in OverThreshold()
161 const float iou_threshold, const int bit_mask_len, in NMSKernel()
253 const float iou_threshold, int* d_selected_indices, int* h_nkeep, in NmsGpu()
551 const Tensor& iou_threshold) { in CheckValidInputs()
608 const Tensor& iou_threshold = context->input(3); in Compute() local
647 const Tensor& iou_threshold = context->input(3); in Compute() local
692 const Tensor& iou_threshold = context->input(3); in Compute() local
|
| D | non_max_suppression_op.cc | 307 const float score_threshold, const float iou_threshold, in DoNMSPerClass()
444 const float score_threshold, const float iou_threshold, in BatchedNonMaxSuppressionOp()
635 const Tensor& iou_threshold = context->input(3); in Compute() local
681 const Tensor& iou_threshold = context->input(3); in Compute() local
737 const Tensor& iou_threshold = context->input(3); in Compute() local
803 const Tensor& iou_threshold = context->input(3); in Compute() local
948 const Tensor& iou_threshold = context->input(4); in Compute() local
|
| D | non_max_suppression_op_test.cc | 34 void MakeOp(float iou_threshold) { in MakeOp()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | non_max_suppression_test.cc | 70 explicit NonMaxSuppressionV4OpModel(const float iou_threshold, in NonMaxSuppressionV4OpModel()
160 explicit NonMaxSuppressionV5OpModel(const float iou_threshold, in NonMaxSuppressionV5OpModel()
|
| D | non_max_suppression.cc | 222 const float iou_threshold = *GetTensorData<float>(input_iou_threshold); in Eval() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_impl.py | 3649 iou_threshold=0.5, argument
3702 iou_threshold=0.5, argument
4957 iou_threshold=0.5, argument
5082 def _self_suppression(iou, _, iou_sum, iou_threshold): argument
5120 def _cross_suppression(boxes, box_slice, iou_threshold, inner_idx, tile_size): argument
5148 def _suppression_loop_body(boxes, iou_threshold, output_size, idx, tile_size): argument
5170 def cross_suppression_func(boxes, box_slice, iou_threshold, inner_idx): argument
5221 iou_threshold=0.5, argument
5326 iou_threshold=0.5, argument
5513 def suppression_loop_body(boxes, iou_threshold, output_size, idx): argument
[all …]
|
| D | image_ops_test.py | 5121 def func(boxes, scores, max_output_size, iou_threshold): argument
5200 def func(boxes, scores, max_output_size, iou_threshold, score_threshold): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | nms_utils.cc | 48 Value iou_threshold = func_.getArgument(3); in RewriteFunc() local
|
| /external/tensorflow/tensorflow/lite/tools/evaluation/stages/utils/ |
| D | image_metrics.h | 95 float iou_threshold = 0.5; member
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | image_ops.cc | 153 ShapeHandle iou_threshold; in NMSShapeFn() local
176 ShapeHandle iou_threshold; in SoftNMSShapeFn() local
983 ShapeHandle iou_threshold; in __anon30b5031d1a02() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | non_max_suppression.h | 89 const float iou_threshold, in NonMaxSuppression()
|
| /external/tensorflow/tensorflow/lite/tools/evaluation/proto/ |
| D | evaluation_stages.proto | 275 optional float iou_threshold = 1; field
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 3141 auto iou_threshold = in TEST_P() local
3164 const float iou_threshold; in TEST_P() member
|
| D | convert_nodes.cc | 5927 TRT_ShapedWeights iou_threshold = inputs.at(4).weights(); in ConvertCombinedNMS() local
|