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

22 lines
634 B
JSON
Executable File

{
"Batch2Space":{
"parameters":
[
{"name": "block_size",
"dtype": "std::vector<int>"
},
{"name": "crop",
"dtype": "std::vector<int>"
},
{"name": "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"]
}
]
}
}