62386f9bdb92662267f18fcf766c84ef6d0e7124,sonnet/python/modules/conv.py,_ConvND,_apply_mask,#_ConvND#,525

Before Change

          )
      )
    if self._mask.shape != w.shape[:self._mask.shape.ndims]:
      raise base.IncompatibleShapeError(
          "Invalid mask shape: {}. Weight shape: {}".format(
              self._mask.shape, w.shape
          )
      )
    // TF broadcasting is a bit fragile.
    // Expand the shape of self._mask by one dim at a time to the right
    // until the rank matches `weight_shape`.
    while self._mask.shape.ndims < w.shape.ndims:

After Change

          )
      )
    if mask_shape != w_shape[:mask_shape.ndims]:
      raise base.IncompatibleShapeError(
          "Invalid mask shape: {}. Weight shape: {}".format(
              mask_shape, w_shape
          )
      )
    // TF broadcasting is a bit fragile.
    // Expand the shape of self._mask by one dim at a time to the right
    // until the rank matches `weight_shape`.
    while self._mask.get_shape().ndims < w_shape.ndims:

Instances