Skip to content

Instantly share code, notes, and snippets.

@tucan9389
Created May 1, 2023 04:57
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save tucan9389/d4d576fd45e8827c1d49ab7c04518da7 to your computer and use it in GitHub Desktop.
Save tucan9389/d4d576fd45e8827c1d49ab7c04518da7 to your computer and use it in GitHub Desktop.
{
"functions": [
{
"name": "@test_onnx_conv_simple_pattern",
"inputs": [
"%arg0",
"%arg1"
],
"inputTypes": [
"tensor<5x3x32x32xf32>",
"tensor<?x3x2x2xf32>"
],
"outputTypes": {
"0": {
"type": "tensor<5x?x31x31xf32>"
}
},
"operations": [
{
"name": "onnx.NoValue",
"attributes": {
"value": "value"
},
"inputs": [],
"inputTypes": [],
"outputs": [
"%0"
],
"outputTypes": [
"none"
],
"body": null
},
{
"name": "onnx.Conv",
"attributes": {
"auto_pad": "NOTSET",
"kernel_shape": "[ 2 , 2 ]",
"pads": "[ 0 , 0 , 0 , 0 ]"
},
"inputs": [
"%arg0",
"%arg1",
"%0"
],
"inputTypes": [
"tensor<5x3x32x32xf32>",
"tensor<?x3x2x2xf32>",
"none"
],
"outputs": [
"%1"
],
"outputTypes": [
"tensor<5x?x31x31xf32>"
],
"body": null
},
{
"name": "return",
"attributes": {},
"inputs": [
"%1"
],
"inputTypes": [
"tensor<5x?x31x31xf32>"
],
"outputs": [],
"outputTypes": [],
"body": null
}
]
}
],
"operations": [],
"attributes": {}
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment