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

Function GetCudnnFrontendHeurMode

tensorflow/stream_executor/cuda/cuda_dnn.cc:3182–3188  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

3180
3181#if CUDNN_VERSION >= 8100
3182cudnnBackendHeurMode_t GetCudnnFrontendHeurMode() {
3183#if CUDNN_VERSION >= 8200
3184 return CUDNN_HEUR_MODE_B;
3185#else
3186 return CUDNN_HEUR_MODE_INSTANT;
3187#endif // CUDNN_VERSION >= 8200
3188}
3189#endif // CUDNN_VERSION >= 8100
3190
3191dnn::DataType GetConvAccumulatorType(dnn::DataType data_type) {

Callers 2

DoConvolveMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected