Home
last modified time | relevance | path

Searched refs:dnn_support (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc267 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveAlgorithms() local
268 if (!dnn_support) { in GetConvolveAlgorithms()
273 return dnn_support->GetConvolveAlgorithms(with_winograd_nonfused, cc_major, in GetConvolveAlgorithms()
286 dnn::DnnSupport *dnn_support = AsDnn(); in GetMIOpenConvolveAlgorithms() local
287 if (!dnn_support) { in GetMIOpenConvolveAlgorithms()
290 return dnn_support->GetMIOpenConvolveAlgorithms( in GetMIOpenConvolveAlgorithms()
298 dnn::DnnSupport *dnn_support = AsDnn(); in GetRnnAlgorithms() local
299 if (!dnn_support) { in GetRnnAlgorithms()
302 return dnn_support->GetRnnAlgorithms(out_algorithms); in GetRnnAlgorithms()
308 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveBackwardDataAlgorithms() local
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.h165 auto* dnn_support = stream_exec->AsDnn(); in ShouldIncludeWinogradNonfusedAlgo() local
166 if (!dnn_support) { in ShouldIncludeWinogradNonfusedAlgo()
170 auto version = dnn_support->GetVersion(); in ShouldIncludeWinogradNonfusedAlgo()