@@ -36,18 +36,18 @@ def alignToContext(self, ctxt: NetworkContext,
3636
3737// 1D Conv (Name: ${nodeName}, Op: ${nodeOp})
3838BEGIN_SINGLE_CORE
39- ${data_in_type.typeName} ref_${data_out }_${data_in} = ${data_in};
40- ${data_out_type.typeName} ref_${data_out }_${data_out} = ${data_out};
39+ ${data_in_type.typeName} ref_${nodeName }_${data_in} = ${data_in};
40+ ${data_out_type.typeName} ref_${nodeName }_${data_out} = ${data_out};
4141
4242 for (uint32_t n=0; n<${batch}; ++n) {
4343 Conv2d_s${data_in_type.referencedType.typeWidth}_s${weight_type.referencedType.typeWidth}_s${data_out_type.referencedType.typeWidth}_NCHW(
44- ref_${data_out }_${data_in}, ${ch_im_in}, 1, ${dim_im_in_y},
44+ ref_${nodeName }_${data_in}, ${ch_im_in}, 1, ${dim_im_in_y},
4545 ${weight}, ${ch_im_out}, 1, ${dim_kernel_y},
4646 1, ${stride_y},
47- ref_${data_out }_${data_out}, ${input_offset}, ${output_offset}
47+ ref_${nodeName }_${data_out}, ${input_offset}, ${output_offset}
4848 );
49- ref_${data_out }_${data_in} += ${batchOffsetIn};
50- ref_${data_out }_${data_out} += ${batchOffsetOut};
49+ ref_${nodeName }_${data_in} += ${batchOffsetIn};
50+ ref_${nodeName }_${data_out} += ${batchOffsetOut};
5151 }
5252END_SINGLE_CORE
5353""" )
@@ -60,18 +60,18 @@ def alignToContext(self, ctxt: NetworkContext,
6060
6161// 2D Conv (Name: ${nodeName}, Op: ${nodeOp})
6262BEGIN_SINGLE_CORE
63- ${data_in_type.typeName} ref_${data_out }_${data_in} = ${data_in};
64- ${data_out_type.typeName} ref_${data_out }_${data_out} = ${data_out};
63+ ${data_in_type.typeName} ref_${nodeName }_${data_in} = ${data_in};
64+ ${data_out_type.typeName} ref_${nodeName }_${data_out} = ${data_out};
6565
6666 for (uint32_t n=0; n<${batch}; ++n) {
6767 Conv2d_s${data_in_type.referencedType.typeWidth}_s${weight_type.referencedType.typeWidth}_s${data_out_type.referencedType.typeWidth}_NCHW(
68- ref_${data_out }_${data_in}, ${ch_im_in}, ${dim_im_in_x}, ${dim_im_in_y},
68+ ref_${nodeName }_${data_in}, ${ch_im_in}, ${dim_im_in_x}, ${dim_im_in_y},
6969 ${weight}, ${ch_im_out}, ${dim_kernel_x}, ${dim_kernel_y},
7070 ${stride_x}, ${stride_y},
71- ref_${data_out }_${data_out}, ${input_offset}, ${output_offset}
71+ ref_${nodeName }_${data_out}, ${input_offset}, ${output_offset}
7272 );
73- ref_${data_out }_${data_in} += ${batchOffsetIn};
74- ref_${data_out }_${data_out} += ${batchOffsetOut};
73+ ref_${nodeName }_${data_in} += ${batchOffsetIn};
74+ ref_${nodeName }_${data_out} += ${batchOffsetOut};
7575 }
7676END_SINGLE_CORE
7777""" )
0 commit comments