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

19 lines
523 B
JSON
Executable File

{
"BatchNorm":{
"parameters":
[
{"name": "eps",
"dtype": "float"
},
{"name": "input_layout",
"dtype": "tim::vx::DataLayout",
"Optional": "true",
"default": "tim::vx::DataLayout::WHCN",
"range":["tim::vx::DataLayout::ANY",
"tim::vx::DataLayout::WHCN",
"tim::vx::DataLayout::CWHN"]
}
]
}
}