Skip to content

Commit 569cf41

Browse files
authored
Conv1d channel last bug fix (#18259)
Differential Revision: D97009237 Pull Request resolved: #18259
1 parent 7a2e230 commit 569cf41

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

backends/cadence/hifi/operators/op_quantized_conv1d_nlc.cpp

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -47,14 +47,14 @@ void xa_opt_quantized_conv1d_nlc_asym8sxsym8s_asym8s(
4747
(WORD32* __restrict__)bias.const_data_ptr<int32_t>();
4848

4949
WORD32 batches = input.size(0);
50-
WORD32 input_channels = input.size(1);
51-
WORD32 input_width = input.size(2);
50+
WORD32 input_channels = input.size(2);
51+
WORD32 input_width = input.size(1);
5252
WORD32 input_height = 1;
5353
WORD32 kernel_height = 1;
5454
WORD32 out_channels = weight.size(0);
55-
WORD32 kernel_channels = weight.size(1);
56-
WORD32 kernel_width = weight.size(2);
57-
WORD32 out_width = out.size(2);
55+
WORD32 kernel_channels = weight.size(2);
56+
WORD32 kernel_width = weight.size(1);
57+
WORD32 out_width = out.size(1);
5858
WORD32 out_height = 1;
5959
WORD32 x_stride = stride[1];
6060
WORD32 y_stride = stride[0];
@@ -70,7 +70,7 @@ void xa_opt_quantized_conv1d_nlc_asym8sxsym8s_asym8s(
7070

7171
WORD32 out_zero_bias = output_zero_point;
7272

73-
WORD32 out_data_format = 1;
73+
WORD32 out_data_format = 0;
7474

7575
WORD32 p_out_multiplier32[out_channels];
7676
WORD32 p_out_shift32[out_channels];

0 commit comments

Comments
 (0)