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

Before Change

def maxpool(converter: Converter, node: Any, inputs: List[str]) -> Any:
    raise NotImplementedError()

    input = converter.outputs[inputs[0]]

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

After Change

    max = MaxPooling2D(shape, pool_size, strides, padding, channels_first)

    out = max.forward(input)

    return out


def shape(converter: Converter, node: Any, inputs: List[str]) -> Any:
    input = converter.outputs[inputs[0]]

Instances