a120cb4377c48caba2d2dbb25f126a06568e01be,src/graph_transpiler/webdnn/frontend/tensorflow/ops/gen_nn_ops.py,,max_pool_handler,#Any#Any#,331

Before Change

    x = converter.get_variable(tf_op.inputs[0])  // NHWC
    assert tf_op.get_attr("data_format") == b"NHWC"
    x.order.unify(OrderNHWC)

    ksize_nhwc = tf_op.get_attr("ksize")  // type: List[int]
    assert ksize_nhwc[0] == 1
    assert ksize_nhwc[3] == 1

After Change

    data_format = tf_op.get_attr("data_format")
    check_data_format(x, data_format)

    ksize = tuple(tf_op.get_attr("ksize"))  // type: Tuple[int,...]
    assert ksize[x.order.axes_dict[Axis.N]] == 1
    assert ksize[x.order.axes_dict[Axis.C]] == 1
    ksize = (ksize[x.order.axes_dict[Axis.H]], ksize[x.order.axes_dict[Axis.W]])

    stride = tuple(tf_op.get_attr("strides"))  // type: Tuple[int,...]
    assert stride[x.order.axes_dict[Axis.N]] == 1
    assert stride[x.order.axes_dict[Axis.C]] == 1

Instances