62386f9bdb92662267f18fcf766c84ef6d0e7124,sonnet/python/modules/conv_test.py,SharedConvTest,SharedConvTest_1,#,185

Before Change

      (snt.Conv1D, 1, CONV_1D_KWARGS),
      (snt.Conv2D, 2, CONV_2D_KWARGS),
      (snt.Conv3D, 3, CONV_3D_KWARGS),
      (snt.Conv1DTranspose, 1, CONV_1D_TRANSPOSE_KWARGS),
      (snt.Conv2DTranspose, 2, CONV_2D_TRANSPOSE_KWARGS),
      (snt.Conv3DTranspose, 3, CONV_3D_TRANSPOSE_KWARGS),
      (snt.DepthwiseConv2D, 2, DEPTHWISE_CONV_2D_KWARGS),

After Change

      "output_channels": 1,
      "kernel_shape": 3,
  }
  CONV_1D_MASKED_KWARGS = {
      "output_channels": 1,
      "kernel_shape": 3,
      "mask": np.zeros((3, 10, 1), dtype=np.float64),
  }
  SEPARABLE_CONV_1D_KWARGS = {
      "output_channels": 10,
      "channel_multiplier": 1,
      "kernel_shape": 3,

Instances