Home
last modified time | relevance | path

Searched refs:pad_per_class (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DCombinedNonMaxSuppression.pbtxt44 name: "pad_per_class"
94 name: "pad_per_class"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DCombinedNonMaxSuppression.pbtxt44 name: "pad_per_class"
94 name: "pad_per_class"
/external/tensorflow/tensorflow/core/kernels/image/
Dnon_max_suppression_op.cc388 bool pad_per_class, int max_size_per_batch, in SelectResultPerBatch() argument
400 if (!pad_per_class) { in SelectResultPerBatch()
445 bool pad_per_class = false, bool clip_boxes = true) { in BatchedNonMaxSuppressionOp() argument
498 if (pad_per_class) { in BatchedNonMaxSuppressionOp()
515 pad_per_class, max_size_per_class * num_classes, clip_boxes, in BatchedNonMaxSuppressionOp()
Dnon_max_suppression_op_test.cc950 void MakeOp(bool pad_per_class = false, bool clip_boxes = true) { in MakeOp() argument
959 .Attr("pad_per_class", pad_per_class) in MakeOp()
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dcombined_nms_test.py163 pad_per_class=False,
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_CombinedNonMaxSuppression.pbtxt48 name: "pad_per_class"
/external/tensorflow/tensorflow/core/ops/
Dimage_ops.cc238 bool pad_per_class; in CombinedNMSShapeFn() local
239 TF_RETURN_IF_ERROR(c->GetAttr("pad_per_class", &pad_per_class)); in CombinedNMSShapeFn()
240 if (!pad_per_class) { in CombinedNMSShapeFn()
Dops.pbtxt7992 name: "pad_per_class"
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.image.pbtxt37 …lass\', \'max_total_size\', \'iou_threshold\', \'score_threshold\', \'pad_per_class\', \'clip_boxe…
Dtensorflow.raw_ops.pbtxt789 …lass\', \'max_total_size\', \'iou_threshold\', \'score_threshold\', \'pad_per_class\', \'clip_boxe…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.image.pbtxt37 …lass\', \'max_total_size\', \'iou_threshold\', \'score_threshold\', \'pad_per_class\', \'clip_boxe…
Dtensorflow.raw_ops.pbtxt789 …lass\', \'max_total_size\', \'iou_threshold\', \'score_threshold\', \'pad_per_class\', \'clip_boxe…
/external/tensorflow/tensorflow/python/ops/
Dimage_ops_impl.py4959 pad_per_class=False, argument
5033 score_threshold, pad_per_class, clip_boxes)
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes_test.cc3133 bool pad_per_class = false) -> NodeDef { in TEST_P() argument
3146 .PadPerClass(pad_per_class) in TEST_P()
3166 bool pad_per_class; in TEST_P() member
3295 auto node_def = get_nms_nodedef(tf_type_, p.clip_boxes, p.pad_per_class); in TEST_P()
Dconvert_nodes.cc6008 const bool pad_per_class = attrs.get<bool>("pad_per_class"); in ConvertCombinedNMS() local
6011 if (pad_per_class) { in ConvertCombinedNMS()
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v2.pbtxt13220 name: "pad_per_class"