dml_convolution_3d.json

{
    "$schema": "./_schema.json",
    
    "resources": 
    {
        "input": 
        {
            "initialValuesDataType": "FLOAT32",
            "initialValues": { "valueCount": 27, "valueStart": 1, "valueDelta": 0 }
        },
        "filter": 
        {
            "initialValuesDataType": "FLOAT32",
            "initialValues": [0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5, 0.5]
        },
        "output": 
        {
            "initialValuesDataType": "FLOAT32",
            "initialValues": { "valueCount": 8, "value": 0 }
        }
    },

    "dispatchables": 
    {
        "conv3d": 
        {
            "type": "DML_OPERATOR_CONVOLUTION",
            "desc": 
            {
                "InputTensor": { "DataType": "FLOAT32", "Sizes": [1,1,3,3,3] },
                "FilterTensor": { "DataType": "FLOAT32", "Sizes": [1,1,2,2,2] },
                "OutputTensor": { "DataType": "FLOAT32", "Sizes": [1,1,2,2,2] },
                "Mode": "DML_CONVOLUTION_MODE_CROSS_CORRELATION",
                "Direction": "DML_CONVOLUTION_DIRECTION_FORWARD",
                "DimensionCount": 3,
                "Strides": [1,1,1],
                "Dilations": [1,1,1],
                "StartPadding": [0,0,0],
                "EndPadding": [0,0,0],
                "OutputPadding": [0,0,0],
                "GroupCount": 1
            }
        }
    },

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