Skip to content

Commit

Permalink
Fix a minor issue with shape infer.
Browse files Browse the repository at this point in the history
  • Loading branch information
liuliu committed Jan 23, 2024
1 parent 2457499 commit 6bbdf00
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion lib/nnc/cmd/convolution/ccv_nnc_convolution.c
Original file line number Diff line number Diff line change
Expand Up @@ -80,7 +80,7 @@ static void _ccv_nnc_conv_transpose_tensor_auto_forw(const ccv_nnc_cmd_param_t c
outputs[0].format = inputs[0].format;
outputs[0].datatype = inputs[0].datatype;
// Get the channel output from the weight matrix.
const int count = ccv_nnc_tensor_get_n(inputs[1]);
const int count = ccv_nnc_tensor_get_c(inputs[1]);
assert(count == cmd.convolution_transpose.count);
ccv_nnc_tensor_set_c(outputs, ccv_nnc_tensor_nd(inputs[0].dim), count);
ccv_nnc_tensor_set_n(outputs, ccv_nnc_tensor_get_n(inputs[0]));
Expand Down

0 comments on commit 6bbdf00

Please sign in to comment.