Searched refs:heatmap1 (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/runtime/test/generated/models/ |
| D | heatmap_max_keypoint.model.cpp | 353 auto heatmap1 = model->addOperand(&type5); in CreateModel_nhwc_2() local
361 …model->addOperation(ANEURALNETWORKS_HEATMAP_MAX_KEYPOINT, {heatmap1, boxes1, layout}, {score1, key… in CreateModel_nhwc_2()
364 {heatmap1, boxes1}, in CreateModel_nhwc_2()
380 auto heatmap1 = model->addOperand(&type5); in CreateModel_nhwc_relaxed_2() local
388 …model->addOperation(ANEURALNETWORKS_HEATMAP_MAX_KEYPOINT, {heatmap1, boxes1, layout}, {score1, key… in CreateModel_nhwc_relaxed_2()
391 {heatmap1, boxes1}, in CreateModel_nhwc_relaxed_2()
409 auto heatmap1 = model->addOperand(&type23); in CreateModel_nhwc_float16_2() local
417 …model->addOperation(ANEURALNETWORKS_HEATMAP_MAX_KEYPOINT, {heatmap1, boxes1, layout}, {score1, key… in CreateModel_nhwc_float16_2()
420 {heatmap1, boxes1}, in CreateModel_nhwc_float16_2()
437 auto heatmap1 = model->addOperand(&type26); in CreateModel_nhwc_quant8() local
[all …]
|
| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | heatmap_max_keypoint.mod.py | 20 heatmap1 = Input("heatmap", "TENSOR_FLOAT32", "{6, 4, 4, 1}") variable
24 Model().Operation("HEATMAP_MAX_KEYPOINT", heatmap1, boxes1, layout).To(score1, keypoint1)
28 heatmap1: [
78 }).AddNchw(heatmap1, layout).AddVariations("relaxed", "float16")
|