| 118 | |
| 119 | template <> |
| 120 | KernelGenRet opr_fill_attr<megdnn::Argsort>( |
| 121 | std::unordered_map<std::string, CCAttr>& attr_map, megdnn::Argsort* opr, |
| 122 | const TensorNDArray& tensors, KernelGen::Arch arch, |
| 123 | const std::unordered_map<std::string, CCAttr>& proxy_attr) { |
| 124 | auto param = opr->param(); |
| 125 | attr_map["order"] = CCAttr(dnnparam_2_str(param.order)); |
| 126 | return KernelGen::KernelPack::GetKernel(KernType::ArgSortKernel, arch); |
| 127 | } |
| 128 | |
| 129 | template <> |
| 130 | KernelGenRet opr_fill_attr<megdnn::TopK>( |