| 318 | |
| 319 | template <> |
| 320 | KernelGenRet opr_fill_attr<megdnn::BatchedMatrixMulForward>( |
| 321 | std::unordered_map<std::string, CCAttr>& attr_map, |
| 322 | megdnn::BatchedMatrixMulForward* opr, const TensorNDArray& tensors, |
| 323 | KernelGen::Arch arch, |
| 324 | const std::unordered_map<std::string, CCAttr>& proxy_attr) { |
| 325 | auto param = opr->param(); |
| 326 | FILL_MAP(attr_map, param, transposeA); |
| 327 | FILL_MAP(attr_map, param, transposeB); |
| 328 | |
| 329 | FILL_MAP_EX(attr_map, param, format, dnnparam_2_str); |
| 330 | FILL_MAP_EX(attr_map, param, compute_mode, dnnparam_2_str); |
| 331 | return KernelGen::KernelPack::GetKernel(KernType::BatchMatmulKernel, arch); |
| 332 | } |
| 333 | |
| 334 | template <> |
| 335 | KernelGenRet opr_fill_attr<megdnn::WarpPerspectiveForward>( |
nothing calls this directly
no outgoing calls
no test coverage detected