MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / cudnn_conv_from_enum

Method cudnn_conv_from_enum

dnn/src/cuda/conv_bias/algo.cpp:387–395  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

385}
386
387ConvBiasForwardImpl::AlgoBase* ConvBiasForwardImpl::AlgoPack::cudnn_conv_from_enum(
388 cudnnConvolutionFwdAlgo_t algo) {
389 for (auto&& i : cudnn_convs) {
390 if (i.cudnn_enum() == algo)
391 return &i;
392 }
393 megdnn_throw(ssprintf(
394 "can not find cudnn conv fwd algorithm %d", static_cast<int>(algo)));
395}
396
397ConvBiasForwardImpl::AlgoBase* ConvBiasForwardImpl::AlgoPack::
398 cudnn_conv_bias_act_from_enum(cudnnConvolutionFwdAlgo_t algo) {

Callers 1

Calls 1

cudnn_enumMethod · 0.45

Tested by

no test coverage detected