Home
last modified time | relevance | path

Searched defs: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
286 dnn::DnnSupport *dnn_support = AsDnn(); in GetMIOpenConvolveAlgorithms() local
298 dnn::DnnSupport *dnn_support = AsDnn(); in GetRnnAlgorithms() local
308 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveBackwardDataAlgorithms() local
321 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveBackwardFilterAlgorithms() local
372 dnn::DnnSupport *dnn_support = AsDnn(); in createRnnDescriptor() local
387 dnn::DnnSupport *dnn_support = AsDnn(); in createRnnSequenceTensorDescriptor() local
401 dnn::DnnSupport *dnn_support = AsDnn(); in createRnnSequenceTensorDescriptor() local
415 dnn::DnnSupport *dnn_support = AsDnn(); in createRnnStateTensorDescriptor() local
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.h165 auto* dnn_support = stream_exec->AsDnn(); in ShouldIncludeWinogradNonfusedAlgo() local