From a6d12de4064574ae2e4c2a4a5103e0715f6529e5 Mon Sep 17 00:00:00 2001 From: David Huang Date: Thu, 28 Nov 2024 05:39:51 +0000 Subject: [PATCH] fix comment --- experimental/torch_xla2/torch_xla2/ops/jaten.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/experimental/torch_xla2/torch_xla2/ops/jaten.py b/experimental/torch_xla2/torch_xla2/ops/jaten.py index 243189dd0ee..cf5928eada1 100644 --- a/experimental/torch_xla2/torch_xla2/ops/jaten.py +++ b/experimental/torch_xla2/torch_xla2/ops/jaten.py @@ -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:])