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

{
    "Moments":{
        "parameters":
            [
                {"name": "axis",
                 "dtype": "std::vector<int32_t>"
                },
                {"name": "keep_dims",
                 "dtype": "bool",
                 "Optional": "true",
                 "default": "false"
                }
            ]
    }
}