TIM-VX/include/tim/vx/ops/argmax.json

{
    "ArgMax":{
        "parameters":
            [
                {"name":"axis",
                    "dtype": "int32_t"
                }
            ]
    }
}