47462402c76719b29b043dd8b4f4e785254730ab,onnx_tf/handlers/backend/pool_mixin.py,PoolMixin,pool_v10,#Any#Any#Any#Any#Any#Any#,121

Before Change

  @classmethod
  def pool_v10(cls, node, input_dict, pool_func, pooling_type, strict=True):
    x = input_dict[node.inputs[0]]
    x_rank = len(x.get_shape())
    spatial_size = x_rank - 2

    storage_format, _ = get_data_format(x_rank)

    kernel_shape = node.attrs["kernel_shape"]

After Change

    if spatial_size > 3:
      exception.OP_UNSUPPORTED_EXCEPT(
          "MaxPool with {}D input".format(x_rank), "Tensorflow")
    if pooling_type == "MAX_WITH_ARGMAX" and x_rank != 4:
      exception.OP_UNSUPPORTED_EXCEPT(
          "MaxPool with {}D input".format(x_rank), "Tensorflow")
    if node.attrs.get("storage_order", 0) != 0:
      exception.OP_UNSUPPORTED_EXCEPT("MaxPool with column major",
                                      "Tensorflow")

Instances