def _argmax(converter, node, inputs):
    x_in = converter.outputs[inputs[0]]
    axis = converter.outputs[inputs[1]].attr['value'].tensor.int_val[0]
    return converter.protocol.argmax(x_in, axis=axis)
def _argmax(converter, node, inputs):
    x_in = converter.outputs[inputs[0]]
    axis = converter.outputs[inputs[1]].attr['value'].tensor.int_val[0]
    return converter.protocol.argmax(x_in, axis=axis)