Searched refs:Iou (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | nn_detect_ops_declare.cc | 48 INPUT_MAP(Iou) = {{1, INPUT_DESC(bboxes)}, {2, INPUT_DESC(gtboxes)}};
49 ATTR_MAP(Iou) = {{"mode", ATTR_DESC(mode, AnyTraits<std::string>())}};
50 OUTPUT_MAP(Iou) = {{0, OUTPUT_DESC(overlap)}};
51 REG_ADPT_DESC(Iou, kNameIOU, ADPT_DESC(Iou))
|
| D | nn_detect_ops_declare.h | 26 DECLARE_OP_ADAPTER(Iou)
27 DECLARE_OP_USE_OUTPUT(Iou)
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3_darknet53/src/ |
| D | yolo.py | 240 class Iou(nn.Cell): class
243 super(Iou, self).__init__()
293 self.iou = Iou()
|
| /third_party/mindspore/tests/st/model_zoo_tests/yolov3/src/ |
| D | yolov3.py | 455 class Iou(nn.Cell): class
458 super(Iou, self).__init__()
515 self.iou = Iou()
|