39 lines
1.4 KiB
JSON
Executable File
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"]
|
|
}
|
|
|
|
]
|
|
}
|
|
} |