Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc251 cudnnConvolutionFwdAlgo_t ToConvForwardAlgo(dnn::AlgorithmDesc algorithm) { in ToConvForwardAlgo() function
2909 /*algo=*/ToConvForwardAlgo(algorithm_desc), in AllocateCudnnConvolutionForwardWorkspace()
4036 /*algo=*/ToConvForwardAlgo(algo), in Create()
4154 /*algo=*/ToConvForwardAlgo(algo), in operator ()()
4898 /*algo=*/ToConvForwardAlgo(algo), in Create()
4981 /*convDesc=*/conv_.handle(), ToConvForwardAlgo(algo), in operator ()()
4990 VLOG(4) << "conv with algorithm " << ToConvForwardAlgo(algo) in operator ()()
5003 VLOG(4) << "conv with algorithm " << ToConvForwardAlgo(algo) in operator ()()