Searched refs:heatmap2 (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | heatmap_max_keypoint.mod.py | 82 heatmap2 = Input("heatmap", "TENSOR_FLOAT32", "{2, 4, 4, 4}") variable
86 Model().Operation("HEATMAP_MAX_KEYPOINT", heatmap2, boxes2, layout).To(score2, keypoint2)
89 heatmap2: ("TENSOR_QUANT8_ASYMM", 0.01, 128),
97 heatmap2: [
131 }).AddNchw(heatmap2, layout).AddVariations("relaxed", "float16", quant8)
|
| /frameworks/ml/nn/runtime/test/generated/models/ |
| D | heatmap_max_keypoint.model.cpp | 804 auto heatmap2 = model->addOperand(&type32); in CreateModel_nhwc_quant8_2() local
812 …model->addOperation(ANEURALNETWORKS_HEATMAP_MAX_KEYPOINT, {heatmap2, boxes2, layout}, {score2, key… in CreateModel_nhwc_quant8_2()
815 {heatmap2, boxes2}, in CreateModel_nhwc_quant8_2()
832 auto heatmap2 = model->addOperand(&type36); in CreateModel_nchw_quant8_2() local
840 …model->addOperation(ANEURALNETWORKS_HEATMAP_MAX_KEYPOINT, {heatmap2, boxes2, layout}, {score2, key… in CreateModel_nchw_quant8_2()
843 {heatmap2, boxes2}, in CreateModel_nchw_quant8_2()
860 auto heatmap2 = model->addOperand(&type32); in CreateModel_dynamic_output_shape_nhwc_quant8_2() local
868 …model->addOperation(ANEURALNETWORKS_HEATMAP_MAX_KEYPOINT, {heatmap2, boxes2, layout}, {score2, key… in CreateModel_dynamic_output_shape_nhwc_quant8_2()
871 {heatmap2, boxes2}, in CreateModel_dynamic_output_shape_nhwc_quant8_2()
888 auto heatmap2 = model->addOperand(&type36); in CreateModel_dynamic_output_shape_nchw_quant8_2() local
[all …]
|