MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / GetConvolveAlgorithms

Method GetConvolveAlgorithms

tensorflow/stream_executor/rocm/rocm_dnn.cc:3002–3015  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

3000}
3001
3002bool MIOpenSupport::GetConvolveAlgorithms(
3003 // ROCM TODO: refactor cc_major / cc_minor
3004 bool with_winograd_nonfused, int cc_major, int cc_minor,
3005 std::vector<dnn::AlgorithmDesc>* out_algorithms) {
3006 out_algorithms->assign({
3007 // clang-format off
3008 dnn::AlgorithmDesc(miopenConvolutionFwdAlgoGEMM, false),
3009 dnn::AlgorithmDesc(miopenConvolutionFwdAlgoDirect, false),
3010 dnn::AlgorithmDesc(miopenConvolutionFwdAlgoFFT, false),
3011 dnn::AlgorithmDesc(miopenConvolutionFwdAlgoWinograd, false),
3012 // clang-format on
3013 });
3014 return true;
3015}
3016
3017bool MIOpenSupport::GetRnnAlgorithms(
3018 std::vector<dnn::AlgorithmDesc>* out_algorithms) {

Callers

nothing calls this directly

Calls 2

AlgorithmDescClass · 0.50
assignMethod · 0.45

Tested by

no test coverage detected