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

39 lines
1.4 KiB
JSON
Executable File

{
"Pool2d":{
"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":"pad",
"dtype":"std::array<uint32_t, 4>"
},
{"name":"ksize",
"dtype":"std::array<uint32_t, 2>"
},
{"name":"stride",
"dtype":"std::array<uint32_t, 2>"
},
{"name":"round_type",
"dtype":"tim::vx::RoundType",
"Optional":"true",
"default":"tim::vx::RoundType::FLOOR",
"range": ["tim::vx::RoundType::FLOOR",
"tim::vx::RoundType::CEILING"]
},
{"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"]
}
]
}
}