| /external/tensorflow/tensorflow/core/kernels/ |
| D | non_max_suppression_op.cc | 108 typename TTypes<T, 2>::ConstTensor boxes, int i, int j, T iou_threshold) { in IOUGreaterThanThreshold() argument
128 return iou > iou_threshold; in IOUGreaterThanThreshold()
229 const float score_threshold, const float iou_threshold, in BatchedNonMaxSuppressionOp() argument
332 selected[j], iou_threshold)) { in BatchedNonMaxSuppressionOp()
496 const Tensor& iou_threshold = context->input(3); in Compute() local
497 OP_REQUIRES(context, TensorShapeUtils::IsScalar(iou_threshold.shape()), in Compute()
499 iou_threshold.shape().DebugString())); in Compute()
500 const float iou_threshold_val = iou_threshold.scalar<float>()(); in Compute()
537 const Tensor& iou_threshold = context->input(3); in Compute() local
538 OP_REQUIRES(context, TensorShapeUtils::IsScalar(iou_threshold.shape()), in Compute()
[all …]
|
| D | non_max_suppression_op.h | 30 float iou_threshold, float score_threshold,
|
| D | non_max_suppression_op_test.cc | 34 void MakeOp(float iou_threshold) { in MakeOp() argument
39 .Attr("iou_threshold", iou_threshold) in MakeOp()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | image_ops_test.py | 747 iou_threshold = array_ops.placeholder(iou_threshold_np.dtype,
756 iou_threshold=iou_threshold,
763 iou_threshold: iou_threshold_np
785 iou_threshold = array_ops.placeholder(iou_threshold_np.dtype,
794 iou_threshold=iou_threshold,
801 iou_threshold: iou_threshold_np
827 iou_threshold = array_ops.placeholder(iou_threshold_np.dtype,
836 iou_threshold=iou_threshold,
842 iou_threshold: iou_threshold_np,
870 iou_threshold = array_ops.placeholder(iou_threshold_np.dtype,
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_NonMaxSuppressionV2.pbtxt | 24 name: "iou_threshold"
55 boxes, scores, max_output_size, iou_threshold)
|
| D | api_def_NonMaxSuppression.pbtxt | 31 name: "iou_threshold"
53 boxes, scores, max_output_size, iou_threshold)
|
| D | api_def_NonMaxSuppressionV3.pbtxt | 24 name: "iou_threshold"
61 boxes, scores, max_output_size, iou_threshold, score_threshold)
|
| D | api_def_NonMaxSuppressionV4.pbtxt | 24 name: "iou_threshold"
75 boxes, scores, max_output_size, iou_threshold, score_threshold)
|
| D | api_def_CombinedNonMaxSuppression.pbtxt | 32 name: "iou_threshold"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_impl.py | 2444 iou_threshold=0.5, argument
2485 iou_threshold = ops.convert_to_tensor(iou_threshold, name='iou_threshold')
2489 iou_threshold, score_threshold)
2496 iou_threshold=0.5, argument
2540 iou_threshold = ops.convert_to_tensor(iou_threshold, name='iou_threshold')
2545 boxes, scores, max_output_size, iou_threshold, score_threshold,
2549 boxes, scores, max_output_size, iou_threshold, score_threshold)
3483 iou_threshold=0.5, argument
3537 iou_threshold = ops.convert_to_tensor(
3538 iou_threshold, dtype=dtypes.float32, name='iou_threshold')
[all …]
|
| D | image_ops_test.py | 4298 iou_threshold = constant_op.constant(iou_threshold_np)
4300 boxes, scores, max_output_size, iou_threshold).eval()
4361 iou_threshold = constant_op.constant(iou_threshold_np)
4363 boxes, scores, max_output_size, iou_threshold).eval()
4371 iou_threshold = constant_op.constant(iou_threshold_np)
4373 boxes, scores, max_output_size, iou_threshold).eval()
4382 iou_threshold = constant_op.constant(iou_threshold_np)
4384 boxes, scores, max_output_size, iou_threshold, score_threshold)
4401 iou_threshold = constant_op.constant(iou_threshold_np)
4407 iou_threshold,
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.image.pbtxt | 37 …s\', \'scores\', \'max_output_size_per_class\', \'max_total_size\', \'iou_threshold\', \'score_thr…
125 …argspec: "args=[\'boxes\', \'scores\', \'max_output_size\', \'iou_threshold\', \'score_threshold\'…
133 …argspec: "args=[\'boxes\', \'scores\', \'max_output_size\', \'iou_threshold\', \'score_threshold\'…
|
| D | tensorflow.raw_ops.pbtxt | 597 …s\', \'scores\', \'max_output_size_per_class\', \'max_total_size\', \'iou_threshold\', \'score_thr…
2145 …argspec: "args=[\'boxes\', \'scores\', \'max_output_size\', \'iou_threshold\', \'name\'], varargs=…
2149 …argspec: "args=[\'boxes\', \'scores\', \'max_output_size\', \'iou_threshold\', \'name\'], varargs=…
2153 …argspec: "args=[\'boxes\', \'scores\', \'max_output_size\', \'iou_threshold\', \'score_threshold\'…
2157 …argspec: "args=[\'boxes\', \'scores\', \'max_output_size\', \'iou_threshold\', \'score_threshold\'…
|
| D | tensorflow.pbtxt | 505 …s\', \'scores\', \'max_output_size_per_class\', \'max_total_size\', \'iou_threshold\', \'score_thr…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.image.pbtxt | 37 …s\', \'scores\', \'max_output_size_per_class\', \'max_total_size\', \'iou_threshold\', \'score_thr…
125 …argspec: "args=[\'boxes\', \'scores\', \'max_output_size\', \'iou_threshold\', \'score_threshold\'…
133 …argspec: "args=[\'boxes\', \'scores\', \'max_output_size\', \'iou_threshold\', \'score_threshold\'…
|
| D | tensorflow.raw_ops.pbtxt | 597 …s\', \'scores\', \'max_output_size_per_class\', \'max_total_size\', \'iou_threshold\', \'score_thr…
2145 …argspec: "args=[\'boxes\', \'scores\', \'max_output_size\', \'iou_threshold\', \'name\'], varargs=…
2149 …argspec: "args=[\'boxes\', \'scores\', \'max_output_size\', \'iou_threshold\', \'name\'], varargs=…
2153 …argspec: "args=[\'boxes\', \'scores\', \'max_output_size\', \'iou_threshold\', \'score_threshold\'…
2157 …argspec: "args=[\'boxes\', \'scores\', \'max_output_size\', \'iou_threshold\', \'score_threshold\'…
|
| D | tensorflow.pbtxt | 965 …s\', \'scores\', \'max_output_size_per_class\', \'max_total_size\', \'iou_threshold\', \'score_thr…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | image_ops.cc | 119 ShapeHandle iou_threshold; in NMSShapeFn() local
120 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &iou_threshold)); in NMSShapeFn()
814 ShapeHandle iou_threshold; in __anonc970e8b61702() local
815 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &iou_threshold)); in __anonc970e8b61702()
|
| D | ops.pbtxt | 5722 name: "iou_threshold"
20852 name: "iou_threshold"
20874 name: "iou_threshold"
20910 name: "iou_threshold"
20950 name: "iou_threshold"
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 19619 …*Scope, boxes tf.Output, scores tf.Output, max_output_size tf.Output, iou_threshold tf.Output) (se…
19626 boxes, scores, max_output_size, iou_threshold,
37733 …*Scope, boxes tf.Output, scores tf.Output, max_output_size tf.Output, iou_threshold tf.Output, sco…
37740 boxes, scores, max_output_size, iou_threshold, score_threshold,
37795 …*Scope, boxes tf.Output, scores tf.Output, max_output_size tf.Output, iou_threshold tf.Output, sco…
37806 boxes, scores, max_output_size, iou_threshold, score_threshold,
37891 …utput, max_output_size_per_class tf.Output, max_total_size tf.Output, iou_threshold tf.Output, sco…
37902 boxes, scores, max_output_size_per_class, max_total_size, iou_threshold, score_threshold,
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v1.pbtxt | 13254 name: "iou_threshold"
41941 name: "iou_threshold"
41963 name: "iou_threshold"
41986 name: "iou_threshold"
42022 name: "iou_threshold"
42049 name: "iou_threshold"
42089 name: "iou_threshold"
42127 name: "iou_threshold"
|
| D | ops_history.v2.pbtxt | 13196 name: "iou_threshold"
41418 name: "iou_threshold"
41440 name: "iou_threshold"
41463 name: "iou_threshold"
41499 name: "iou_threshold"
41526 name: "iou_threshold"
41566 name: "iou_threshold"
41604 name: "iou_threshold"
|
| D | ops_history.v0.pbtxt | 18930 name: "iou_threshold"
|