dml_reduce.json

{
    "$schema": "./_schema.json",
    
    "resources": 
    {
        "input": 
        {
            "initialValuesDataType": "FLOAT32",
            "initialValues": { "valueCount": 9, "valueStart": 1, "valueDelta": 1 }
        },
        "output": 
        {
            "initialValuesDataType": "FLOAT32",
            "initialValues": { "valueCount": 3, "value": 0 }
        }
    },

    "dispatchables": 
    {
        "sum": 
        {
            "type": "DML_OPERATOR_REDUCE",
            "desc": 
            {
                "InputTensor": { "DataType": "FLOAT32", "Sizes": [1,1,3,3] },
                "OutputTensor": { "DataType": "FLOAT32", "Sizes": [1,1,3,1] },
                "Function": "DML_REDUCE_FUNCTION_SUM",
                "AxisCount": 1,
                "Axes": [3]
            }
        }
    },

    "commands": 
    [
        {
            "type": "dispatch",
            "dispatchable": "sum",
            "bindings": 
            {
                "InputTensor": "input",
                "OutputTensor": "output"
            }
        },
        { "type": "print", "resource": "input" },
        { "type": "print", "resource": "output" }
    ]
}