de769df64ad870d9cc594c09f2d2e4cd5ae0ea21,sonnet/python/modules/conv.py,DepthwiseConv2D,_build,#DepthwiseConv2D#Any#,2321

Before Change

          "Input Tensor must have shape (batch_size, input_height, "
          "input_width, input_channels)")

    if self._data_format == DATA_FORMAT_NCHW:
      input_channels = self._input_shape[1]
    else:
      input_channels = self._input_shape[3]

    if input_channels is None:
      raise base.UnderspecifiedError(
          "Number of input channels must be known at module build time")

After Change

    
    // Handle input whose shape is unknown during graph creation.
    self._input_shape = tuple(inputs.get_shape().as_list())
    if len(self._input_shape) != len(self._data_format):
      raise base.IncompatibleShapeError((
          "Input Tensor must have rank {} corresponding to "
          "data_format {}, but instead was {}.").format(
              len(self._data_format), self._data_format, self._input_shape))

    self._input_channels = self._input_shape[self._channel_index]
    if self._input_channels is None:
      raise base.UnderspecifiedError(

Instances