Skip to content

Commit

Permalink
fix comment
Browse files Browse the repository at this point in the history
  • Loading branch information
dvhg committed Nov 28, 2024
1 parent 92bcb82 commit a6d12de
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion experimental/torch_xla2/torch_xla2/ops/jaten.py
Original file line number Diff line number Diff line change
Expand Up @@ -1068,7 +1068,7 @@ def create_default_conv_dimension_numbers(num_spatial_dims):
if transposed:
rhs = jnp.flip(weight, range(2, 1+num_shape_dim))
if groups != 1:
# reshape kernels for tranposed depthwise convolution
# reshape filters for tranposed depthwise convolution
assert rhs.shape[0] % groups == 0
rhs_shape = [rhs.shape[0]//groups, rhs.shape[1]*groups]
rhs_shape.extend(rhs.shape[2:])
Expand Down

0 comments on commit a6d12de

Please sign in to comment.