Searched refs:dnn_support (Results 1 – 2 of 2) sorted by relevance
326 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveAlgorithms() local327 if (!dnn_support) { in GetConvolveAlgorithms()332 return dnn_support->GetConvolveAlgorithms(with_winograd_nonfused, cc_major, in GetConvolveAlgorithms()338 dnn::DnnSupport *dnn_support = AsDnn(); in GetRnnAlgorithms() local339 if (!dnn_support) { in GetRnnAlgorithms()342 return dnn_support->GetRnnAlgorithms(out_algorithms); in GetRnnAlgorithms()348 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveBackwardDataAlgorithms() local349 if (!dnn_support) { in GetConvolveBackwardDataAlgorithms()354 return dnn_support->GetConvolveBackwardDataAlgorithms( in GetConvolveBackwardDataAlgorithms()361 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveBackwardFilterAlgorithms() local[all …]
145 auto* dnn_support = stream_exec->AsDnn(); in ShouldIncludeWinogradNonfusedAlgo() local146 if (!dnn_support) { in ShouldIncludeWinogradNonfusedAlgo()150 auto version = dnn_support->GetVersion(); in ShouldIncludeWinogradNonfusedAlgo()