2023-06-02 07:51:10 +08:00
|
|
|
{
|
|
|
|
|
"RoiPool":{
|
|
|
|
|
"parameters":
|
|
|
|
|
[
|
|
|
|
|
{"name":"type",
|
|
|
|
|
"dtype": "tim::vx::PoolType",
|
|
|
|
|
"range": ["tim::vx::PoolType::MAX",
|
|
|
|
|
"tim::vx::PoolType::AVG",
|
|
|
|
|
"tim::vx::PoolType::L2",
|
|
|
|
|
"tim::vx::POOLType::AVG_ANDROID"]
|
|
|
|
|
},
|
|
|
|
|
{"name":"scale",
|
|
|
|
|
"dtype": "float"
|
|
|
|
|
},
|
|
|
|
|
{"name":"size",
|
|
|
|
|
"dtype": "std::array<uint32_t, 2>"
|
2023-07-08 23:39:56 +08:00
|
|
|
},
|
|
|
|
|
{"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"]
|
2023-06-02 07:51:10 +08:00
|
|
|
}
|
|
|
|
|
]
|
|
|
|
|
}
|
|
|
|
|
}
|