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

38 lines
1.3 KiB
JSON
Executable File

{
"Pool1d":{
"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, 2>"
},
{"name":"ksize",
"dtype":"uint32_t"
},
{"name":"stride",
"dtype":"uint32_t"
},
{"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::WCN",
"range": ["tim::vx::DataLayout::WCN",
"tim::vx::DataLayout::WIO"]
}
]
}
}