MCPcopy Create free account
hub / github.com/MegEngine/MegCC / ReduceForward>

Method ReduceForward>

compiler/test/kernel/common/src/cc_fill_attr.cpp:365–375  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

363
364template <>
365KernelGenRet opr_fill_attr<megdnn::ReduceForward>(
366 std::unordered_map<std::string, CCAttr>& attr_map, megdnn::ReduceForward* opr,
367 const TensorNDArray& tensors, KernelGen::Arch arch,
368 const std::unordered_map<std::string, CCAttr>& proxy_attr) {
369 auto param = opr->param();
370 FILL_MAP(attr_map, param, axis);
371
372 FILL_MAP_EX(attr_map, param, mode, dnnparam_2_str);
373 FILL_MAP_EX(attr_map, param, data_type, dnnparam_2_str);
374 return KernelGen::KernelPack::GetKernel(KernType::ReduceKernel, arch);
375}
376
377template <>
378KernelGenRet opr_fill_attr<megdnn::CVResize>(

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected