Searched refs:keypoint1 (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/runtime/test/generated/models/ |
| D | heatmap_max_keypoint.model.cpp | 357 auto keypoint1 = model->addOperand(&type7); in CreateModel_nhwc_2() local
361 …ddOperation(ANEURALNETWORKS_HEATMAP_MAX_KEYPOINT, {heatmap1, boxes1, layout}, {score1, keypoint1}); in CreateModel_nhwc_2()
365 {score1, keypoint1}); in CreateModel_nhwc_2()
384 auto keypoint1 = model->addOperand(&type7); in CreateModel_nhwc_relaxed_2() local
388 …ddOperation(ANEURALNETWORKS_HEATMAP_MAX_KEYPOINT, {heatmap1, boxes1, layout}, {score1, keypoint1}); in CreateModel_nhwc_relaxed_2()
392 {score1, keypoint1}); in CreateModel_nhwc_relaxed_2()
413 auto keypoint1 = model->addOperand(&type24); in CreateModel_nhwc_float16_2() local
417 …ddOperation(ANEURALNETWORKS_HEATMAP_MAX_KEYPOINT, {heatmap1, boxes1, layout}, {score1, keypoint1}); in CreateModel_nhwc_float16_2()
421 {score1, keypoint1}); in CreateModel_nhwc_float16_2()
441 auto keypoint1 = model->addOperand(&type27); in CreateModel_nhwc_quant8() local
[all …]
|
| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | heatmap_max_keypoint.mod.py | 23 keypoint1 = Output("keypoint", "TENSOR_FLOAT32", "{6, 1, 2}") variable
24 Model().Operation("HEATMAP_MAX_KEYPOINT", heatmap1, boxes1, layout).To(score1, keypoint1)
70 keypoint1: [
|