Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc231 cudnnConvolutionFwdAlgo_t ToConvForwardAlgo(dnn::AlgorithmDesc algorithm) { in ToConvForwardAlgo() function
2845 /*yDesc=*/output_nd.handle(), /*algo=*/ToConvForwardAlgo(algorithm_desc), in AllocateCudnnConvolutionForwardWorkspace()
3999 /*algo=*/ToConvForwardAlgo(algorithm_desc), in DoConvolve()
4982 /*convDesc=*/conv.handle(), ToConvForwardAlgo(algo_desc), in DoFusedConvolve()
4991 VLOG(4) << "conv with algorithm " << ToConvForwardAlgo(algo_desc) in DoFusedConvolve()
5005 VLOG(4) << "conv with algorithm " << ToConvForwardAlgo(algo_desc) in DoFusedConvolve()