| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | roi_pooling.mod.py | 58 roi2 = Input("roi", "TENSOR_FLOAT32", "{4, 4}") variable
60 Model().Operation("ROI_POOLING", i2, roi2, [0, 0, 3, 3], 2, 3, 4.0, 4.0, layout).To(o2)
64 roi2: ("TENSOR_QUANT16_ASYMM", 0.125, 0),
100 roi2: [
|
| D | roi_align.mod.py | 56 roi2 = Input("roi", "TENSOR_FLOAT32", "{4, 4}") variable
58 Model().Operation("ROI_ALIGN", i2, roi2, [0, 0, 3, 3], 2, 3, 4.0, 4.0, 4, 4, layout).To(o2)
62 roi2: ("TENSOR_QUANT16_ASYMM", 0.125, 0),
98 roi2: [
|
| /frameworks/ml/nn/runtime/test/generated/models/ |
| D | roi_pooling.model.cpp | 1437 auto roi2 = model->addOperand(&type2); in CreateModel_nhwc_3() local
1458 …model->addOperation(ANEURALNETWORKS_ROI_POOLING, {in2, roi2, param10, param11, param12, param13, p… in CreateModel_nhwc_3()
1461 {in2, roi2}, in CreateModel_nhwc_3()
1481 auto roi2 = model->addOperand(&type2); in CreateModel_nhwc_relaxed_3() local
1502 …model->addOperation(ANEURALNETWORKS_ROI_POOLING, {in2, roi2, param10, param11, param12, param13, p… in CreateModel_nhwc_relaxed_3()
1505 {in2, roi2}, in CreateModel_nhwc_relaxed_3()
1527 auto roi2 = model->addOperand(&type14); in CreateModel_nhwc_quant8_3() local
1548 …model->addOperation(ANEURALNETWORKS_ROI_POOLING, {in2, roi2, param10, param11, param12, param13, p… in CreateModel_nhwc_quant8_3()
1551 {in2, roi2}, in CreateModel_nhwc_quant8_3()
1571 auto roi2 = model->addOperand(&type18); in CreateModel_nhwc_float16_3() local
[all …]
|
| D | bbox_graph.model.cpp | 60 auto roi2 = model->addOperand(&type6); in CreateModel_zero_sized() local
119 …batches, param14, param15, param16, param17, param18, param19}, {scores4, roi2, classes, batches1}… in CreateModel_zero_sized()
123 {scores1, scores4, roi2, classes, batches1}); in CreateModel_zero_sized()
189 auto roi2 = model->addOperand(&type6); in CreateModel_zero_sized_relaxed() local
248 …batches, param14, param15, param16, param17, param18, param19}, {scores4, roi2, classes, batches1}… in CreateModel_zero_sized_relaxed()
252 {scores1, scores4, roi2, classes, batches1}); in CreateModel_zero_sized_relaxed()
320 auto roi2 = model->addOperand(&type25); in CreateModel_zero_sized_float16() local
379 …batches, param14, param15, param16, param17, param18, param19}, {scores4, roi2, classes, batches1}… in CreateModel_zero_sized_float16()
383 {scores1, scores4, roi2, classes, batches1}); in CreateModel_zero_sized_float16()
450 auto roi2 = model->addOperand(&type38); in CreateModel_zero_sized_quant8() local
[all …]
|
| D | detection_postprocess.model.cpp | 816 auto roi2 = model->addOperand(&type10); in CreateModel_2() local
856 …model->addOperation(ANEURALNETWORKS_DETECTION_POSTPROCESSING, {scores2, roi2, anchors2, param22, p… in CreateModel_2()
859 {scores2, roi2, anchors2}, in CreateModel_2()
882 auto roi2 = model->addOperand(&type10); in CreateModel_relaxed_2() local
922 …model->addOperation(ANEURALNETWORKS_DETECTION_POSTPROCESSING, {scores2, roi2, anchors2, param22, p… in CreateModel_relaxed_2()
925 {scores2, roi2, anchors2}, in CreateModel_relaxed_2()
950 auto roi2 = model->addOperand(&type23); in CreateModel_float16_2() local
990 …model->addOperation(ANEURALNETWORKS_DETECTION_POSTPROCESSING, {scores2, roi2, anchors2, param22, p… in CreateModel_float16_2()
993 {scores2, roi2, anchors2}, in CreateModel_float16_2()
1016 auto roi2 = model->addOperand(&type10); in CreateModel_dynamic_output_shape_2() local
[all …]
|
| D | roi_align.model.cpp | 1629 auto roi2 = model->addOperand(&type2); in CreateModel_nhwc_3() local
1656 …model->addOperation(ANEURALNETWORKS_ROI_ALIGN, {in2, roi2, param14, param15, param16, param17, par… in CreateModel_nhwc_3()
1659 {in2, roi2}, in CreateModel_nhwc_3()
1679 auto roi2 = model->addOperand(&type2); in CreateModel_nhwc_relaxed_3() local
1706 …model->addOperation(ANEURALNETWORKS_ROI_ALIGN, {in2, roi2, param14, param15, param16, param17, par… in CreateModel_nhwc_relaxed_3()
1709 {in2, roi2}, in CreateModel_nhwc_relaxed_3()
1731 auto roi2 = model->addOperand(&type27); in CreateModel_nhwc_quant8_3() local
1758 …model->addOperation(ANEURALNETWORKS_ROI_ALIGN, {in2, roi2, param14, param15, param16, param17, par… in CreateModel_nhwc_quant8_3()
1761 {in2, roi2}, in CreateModel_nhwc_quant8_3()
1781 auto roi2 = model->addOperand(&type31); in CreateModel_nhwc_float16_3() local
[all …]
|
| /frameworks/ml/nn/common/operations/ |
| D | GenerateProposals.cpp | 145 float getIoUAxisAligned(const float* roi1, const float* roi2) { in getIoUAxisAligned() argument
147 const float area2 = (roi2[2] - roi2[0]) * (roi2[3] - roi2[1]); in getIoUAxisAligned()
148 const float x1 = std::max(roi1[0], roi2[0]); in getIoUAxisAligned()
149 const float x2 = std::min(roi1[2], roi2[2]); in getIoUAxisAligned()
150 const float y1 = std::max(roi1[1], roi2[1]); in getIoUAxisAligned()
151 const float y2 = std::min(roi1[3], roi2[3]); in getIoUAxisAligned()
|