| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | combined_nms_test.py | 36 max_total_size = 3
45 max_total_size, dtype=dtypes.int32, name='max_total_size')
69 max_total_size = 3
73 nmsed_boxes_shape = [batch_size, max_total_size, 4]
74 nmsed_scores_shape = [batch_size, max_total_size]
75 nmsed_classes_shape = [batch_size, max_total_size]
160 max_total_size=max_total_size_tensor,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | CombinedNonMaxSuppression.pbtxt | 16 name: "max_total_size"
66 name: "max_total_size"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | CombinedNonMaxSuppression.pbtxt | 16 name: "max_total_size"
66 name: "max_total_size"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_CombinedNonMaxSuppression.pbtxt | 26 name: "max_total_size"
51 are padded/clipped to `max_total_size`. If true, the
53 `max_size_per_class`*`num_classes`, unless it exceeds `max_total_size` in
54 which case it is clipped to `max_total_size`. Defaults to false.
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op_benchmark_test.cc | 33 Tensor max_total_size(9000); in CombinedNonMaxSuppression() local
42 .Input(test::graph::Constant(g, max_total_size)) in CombinedNonMaxSuppression()
|
| D | non_max_suppression_op.cc | 933 const Tensor& max_total_size = context->input(3); in Compute() local
935 context, TensorShapeUtils::IsScalar(max_total_size.shape()), in Compute()
937 max_total_size.shape().DebugString())); in Compute()
938 const int max_total_size_per_batch = max_total_size.scalar<int>()(); in Compute()
945 << max_total_size.scalar<int>()() << ")"; in Compute()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | image_ops.cc | 204 ShapeHandle max_total_size; in CombinedNMSShapeFn() local
205 TF_RETURN_IF_ERROR(c->WithRank(c->input(3), 0, &max_total_size)); in CombinedNMSShapeFn()
|
| D | ops.pbtxt | 7964 name: "max_total_size"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.image.pbtxt | 37 …argspec: "args=[\'boxes\', \'scores\', \'max_output_size_per_class\', \'max_total_size\', \'iou_th…
|
| D | tensorflow.raw_ops.pbtxt | 789 …argspec: "args=[\'boxes\', \'scores\', \'max_output_size_per_class\', \'max_total_size\', \'iou_th…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.image.pbtxt | 37 …argspec: "args=[\'boxes\', \'scores\', \'max_output_size_per_class\', \'max_total_size\', \'iou_th…
|
| D | tensorflow.raw_ops.pbtxt | 789 …argspec: "args=[\'boxes\', \'scores\', \'max_output_size_per_class\', \'max_total_size\', \'iou_th…
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | image_ops_impl.py | 4956 max_total_size, argument
5029 max_total_size = ops.convert_to_tensor(max_total_size)
5032 boxes, scores, max_output_size_per_class, max_total_size, iou_threshold,
|
| D | image_ops_test.py | 4860 max_total_size = 2**31
4869 max_total_size=max_total_size)
4880 max_total_size = ops.convert_to_tensor(2**31)
4889 max_total_size=max_total_size)
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 3139 auto max_total_size = in TEST_P() local
3151 max_output_size_per_class, max_total_size, iou_threshold, in TEST_P()
3163 const int32 max_total_size; in TEST_P() member
3290 AddTestWeights<int32>("max_total_size", {1}, {p.max_total_size}); in TEST_P()
|
| D | convert_nodes.cc | 5972 int max_total_size = *(static_cast<int*>(total_size.GetValues())); in ConvertCombinedNMS() local
5973 if (max_total_size <= 0) { in ConvertCombinedNMS()
6012 keep_top_k = std::min(max_size_per_class * num_classes, max_total_size); in ConvertCombinedNMS()
6014 keep_top_k = max_total_size; in ConvertCombinedNMS()
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 20413 …es tf.Output, scores tf.Output, max_output_size_per_class tf.Output, max_total_size tf.Output, iou…
20424 boxes, scores, max_output_size_per_class, max_total_size, iou_threshold, score_threshold,
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 13192 name: "max_total_size"
|