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.cc326 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveAlgorithms() local
338 dnn::DnnSupport *dnn_support = AsDnn(); in GetRnnAlgorithms() local
348 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveBackwardDataAlgorithms() local
361 dnn::DnnSupport *dnn_support = AsDnn(); in GetConvolveBackwardFilterAlgorithms() local
387 dnn::DnnSupport *dnn_support = AsDnn(); in createRnnDescriptor() local
402 dnn::DnnSupport *dnn_support = AsDnn(); in createRnnSequenceTensorDescriptor() local
416 dnn::DnnSupport *dnn_support = AsDnn(); in createRnnSequenceTensorDescriptor() local
430 dnn::DnnSupport *dnn_support = AsDnn(); in createRnnStateTensorDescriptor() local
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_gpu.h145 auto* dnn_support = stream_exec->AsDnn(); in ShouldIncludeWinogradNonfusedAlgo() local