dc09af160f9f04c13fd0764b2712176b83538182,sonnet/python/modules/conv.py,SeparableConv2D,__init__,#SeparableConv2D#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#,2222

Before Change

      if not stride[0] == stride[3] == 1:
        raise base.IncompatibleShapeError(
            "Invalid stride: First and last element must be 1.")
      if not (isinstance(stride[1], numbers.Integral) and
              isinstance(stride[2], numbers.Integral)):
        raise base.IncompatibleShapeError(
            "Invalid stride: stride[1] and [2] must be integer.")
      self._stride = tuple(stride)

After Change

    self._kernel_shape = _fill_and_verify_parameter_shape(kernel_shape, 2,
                                                          "kernel")

    if data_format not in SUPPORTED_DATA_FORMATS:
      raise ValueError("Invalid data_format {:s}. Allowed formats "
                       "{:s}".format(data_format, SUPPORTED_DATA_FORMATS))

    self._data_format = data_format

    // We want to support passing native strides akin to [1, m, n, 1]
    if isinstance(stride, collections.Iterable) and len(stride) == 4:
      if self._data_format == DATA_FORMAT_NHWC:

Instances