50f9caba54a250607938b91b6564a76d0157a53f,tensorflow_encrypted/convert/register.py,,maxpool,#Any#Any#Any#,172

Before Change

    input = converter.outputs[inputs[0]]

    return tf.nn.max_pool(input, list(node.attr["ksize"].list.i),
                          list(node.attr["strides"].list.i),
                          node.attr["padding"].s)

After Change

    ksize = node.attr["ksize"].list.i
    s = node.attr["strides"].list.i

    padding = node.attr["padding"].s.decode("ascii")
    pool_size = [ksize[1], ksize[2]]
    strides = [s[1], s[2]]

    shape = [int(i) for i in input.shape]

Instances