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

{
    "Gather":{
        "parameters":
            [
                {"name": "axis",
                 "dtype": "int"
                },
                {"name": "batch_dims",
                 "dtype": "int",
                 "Optional": "true",
                 "default": "0"
                }
            ]
    }
}