611be2ba4a19a1eb3ff21993af1d5ae5870639b9,onnx_tf/handlers/backend/pool_mixin.py,PoolMixin,pool_v10,#Any#Any#Any#Any#Any#Any#,122

Before Change

    // use the old version of the op
    if ((dilation == [1] * spatial_size) and (not ceil_mode)) or \
       (pooling_type[:3] != "MAX"):
      return cls.pool(node, input_dict, pool_func, pooling_type, strict)

    if x_rank != 4:
      exception.OP_UNSUPPORTED_EXCEPT(
          "MaxPool with {}D input".format(x_rank), "Tensorflow")

After Change

    need_trans = storage_format.startswith("NC")
    if need_trans:
      compute_format = "N" + storage_format[2:] + "C"
      x = tf.transpose(x, perm=get_perm_from_formats(storage_format,
                                                     compute_format))

    dp = DilatedPooling(input=x, kernel_shape=kernel_shape, strides=strides,

Instances