@@ -114,13 +114,16 @@ inline void EvalQuantizedPerChannel(TfLiteContext* context, TfLiteNode* node,
114114
115115 data_dims_t input_dims = {
116116 .width = input_width, .height = input_height,
117- .channels = input_depth, 1
117+ .channels = input_depth, . extra = 1
118118 };
119119 data_dims_t output_dims = {
120120 .width = output_width, .height = output_height,
121- .channels = output_depth, 1
121+ .channels = output_depth, .extra = 1
122+ };
123+ data_dims_t filter_dims = {
124+ .width = filter_width, .height = filter_height,
125+ .channels = 0 , .extra = 0
122126 };
123- data_dims_t filter_dims = {.width = filter_width, .height = filter_height, 0 , 0 };
124127 dw_conv_params_t conv_params = {
125128 .in_offset = input_offset, .out_offset = output_offset,
126129 .ch_mult = depth_multiplier,
@@ -254,13 +257,16 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
254257 if (input->type == kTfLiteInt8 ) {
255258 data_dims_t input_dims = {
256259 .width = input_width, .height = input_height,
257- .channels = num_input_channels, 1
260+ .channels = num_input_channels, . extra = 1
258261 };
259262 data_dims_t output_dims = {
260263 .width = output_width, .height = output_height,
261- .channels = output->dims ->data [3 ], 1
264+ .channels = output->dims ->data [3 ], .extra = 1
265+ };
266+ data_dims_t filter_dims = {
267+ .width = filter_width, .height = filter_height,
268+ .channels = 0 , .extra = 0
262269 };
263- data_dims_t filter_dims = {.width = filter_width, .height = filter_height, 0 , 0 };
264270 dw_conv_params_t conv_params = {
265271 .in_offset = 0 , .out_offset = 0 ,
266272 .ch_mult = params.depth_multiplier ,
0 commit comments