| 275 | |
| 276 | template <> |
| 277 | KernelGenRet opr_fill_attr<megdnn::PoolingForward>( |
| 278 | std::unordered_map<std::string, CCAttr>& attr_map, megdnn::PoolingForward* opr, |
| 279 | const TensorNDArray& tensors, KernelGen::Arch arch, |
| 280 | const std::unordered_map<std::string, CCAttr>& proxy_attr) { |
| 281 | auto param = opr->param(); |
| 282 | FILL_MAP(attr_map, param, stride_h); |
| 283 | FILL_MAP(attr_map, param, stride_w); |
| 284 | FILL_MAP(attr_map, param, pad_h); |
| 285 | FILL_MAP(attr_map, param, pad_w); |
| 286 | FILL_MAP(attr_map, param, window_h); |
| 287 | FILL_MAP(attr_map, param, window_w); |
| 288 | |
| 289 | FILL_MAP_EX(attr_map, param, format, dnnparam_2_str); |
| 290 | FILL_MAP_EX(attr_map, param, mode, dnnparam_2_str); |
| 291 | return KernelGen::KernelPack::GetKernel(KernType::PoolingKernel, arch); |
| 292 | } |
| 293 | |
| 294 | template <> |
| 295 | KernelGenRet opr_fill_attr<megdnn::ConcatForward>( |
nothing calls this directly
no outgoing calls
no test coverage detected