1ed54c01c05043c672ec6986529730a1fcf6a337,onnx_tf/handlers/backend/dilated_maxpooling.py,DilatedPooling,dilated_maxpool_with_argmax,#DilatedPooling#Any#,322

Before Change

        // spatial_size != 2
        assert self.spatial_size == 2

        self._pad_input()

        if list(self.dilations) != [1] * self.spatial_size or \
           force_custom_impl:
            new_input = self._reduce_dilations()

After Change

        if list(self.dilations) != [1] * self.spatial_size or \
           force_custom_impl:
            // pad the input
            self._pad_input()

            new_input = self._reduce_dilations()
            kernel_shape = [1] + list(self.kernel_shape) + [1]
            pooled, new_ind = tf.nn.max_pool_with_argmax(
                                    new_input, ksize=kernel_shape,
                                    strides=kernel_shape, padding="VALID")
            new_ind = self._calc_orig_ind(new_ind)
        else:
            if type(self.padding) is list or \
               self.padding.lower() == "same_lower":
                // pad the input
                self._pad_input()

                padding_ = "VALID"
            elif self.padding.lower() == "same_upper":
                padding_ = "SAME"

Instances