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

Method get_algorithm_heuristic

dnn/src/naive/convolution/opr_impl.cpp:256–263  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

254}
255
256ConvolutionForward::Algorithm* ConvolutionForwardImpl::get_algorithm_heuristic(
257 const TensorLayout& /* src */, const TensorLayout& /* filter */,
258 const TensorLayout& /* dst */, size_t /* workspace_limit_in_bytes */,
259 const AlgoAttribute& positive_attr, const AlgoAttribute& negative_attr) {
260 auto algo = static_cast<HandleImpl*>(handle())->default_conv_fwd_algo();
261 algo->check_attribute(positive_attr, negative_attr);
262 return algo;
263}
264
265ConvolutionForward::Algorithm* ConvolutionForwardImpl::get_algorithm_from_desc(
266 const AlgorithmDesc& desc) {

Callers

nothing calls this directly

Calls 2

default_conv_fwd_algoMethod · 0.80
check_attributeMethod · 0.80

Tested by

no test coverage detected