Searched defs:use_cudnn_frontend (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | stream_executor_pimpl.cc | 294 bool use_cudnn_frontend, dnn::ConvolutionKind kind, in GetConvolveRunners()
315 bool use_cudnn_frontend, dnn::ConvolutionKind kind, in GetFusedConvolveRunners()
|
| H A D | dnn.cc | 143 bool use_cudnn_frontend, dnn::ConvolutionKind kind, in GetFusedConvolveRunners()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_conv_algorithm_picker.cc | 110 const GpuConvConfig& config, se::Stream* stream, bool use_cudnn_frontend, in GetAlgorithms()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 4711 bool use_cudnn_frontend, dnn::ConvolutionKind kind, in GetConvolveRunners()
5138 bool use_cudnn_frontend, dnn::ConvolutionKind kind, in GetFusedConvolveRunners()
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.cc | 3147 bool use_cudnn_frontend, dnn::ConvolutionKind kind, in GetConvolveRunners()
|