lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix bug: Conv2dBackward gradBias shape

This commit is contained in:
Yuxuan Han 2025-04-13 23:14:18 +08:00
parent 8a74e9e78a
commit 6407252044
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/jittor/extern/acl/aclops/conv_op.py vendored
View File

@ -133,7 +133,7 @@ class ConvACL(jt.Function):
output_shapes.append(bias.shape)
output_dtypes.append(bias.dtype)
else:
output_shapes.append([1])
output_shapes.append([weight.shape[0]])
output_dtypes.append(x.dtype)
padding = self.padding
stride = self.stride