| 453 | |
| 454 | template <> |
| 455 | KernelGenRet opr_fill_attr<megdnn::ArgmaxForward>( |
| 456 | std::unordered_map<std::string, CCAttr>& attr_map, megdnn::ArgmaxForward* opr, |
| 457 | const TensorNDArray& tensors, KernelGen::Arch arch, |
| 458 | const std::unordered_map<std::string, CCAttr>& proxy_attr) { |
| 459 | auto param = opr->param(); |
| 460 | FILL_MAP(attr_map, param, axis); |
| 461 | return KernelGen::KernelPack::GetKernel(KernType::ArgmaxKernel, arch); |
| 462 | } |
| 463 | template <> |
| 464 | KernelGenRet opr_fill_attr<megdnn::ConvolutionBackwardData>( |
| 465 | std::unordered_map<std::string, CCAttr>& attr_map, |
nothing calls this directly
no outgoing calls
no test coverage detected