869b558268f198de90822e4ea17182dc74b78213,sonnet/python/modules/conv.py,Conv2D,_build,#Conv2D#Any#,315

Before Change

        if mask_shape != self._kernel_shape:
          raise base.IncompatibleShapeError(
              "Invalid mask shape: {}".format(mask_shape))
        mask = np.reshape(self._mask, self._kernel_shape + (1, 1))
      elif mask_rank == 4:
        if mask_shape != tuple(weight_shape):
          raise base.IncompatibleShapeError(
              "Invalid mask shape: {}".format(mask_shape))

After Change

      mask = self._mask
      mask_shape = mask.shape.as_list()
      if len(mask_shape) == 2:
        if mask_shape != list(self._kernel_shape):
          raise base.IncompatibleShapeError(
              "Invalid mask shape: {}".format(tuple(mask_shape)))
        mask = tf.expand_dims(tf.expand_dims(mask, -1), -1)

Instances