Searched refs:UseTensorOps (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2669 port::StatusOr<bool> UseTensorOps(Stream* stream, dnn::DataType type, in UseTensorOps() function
2702 UseTensorOps(stream, element_type, algo_desc)); in GetCudnnConvolutionForwardAlgorithm()
2742 UseTensorOps(stream, element_type, algo_desc)); in GetCudnnConvolutionForwardAlgorithm()
2764 UseTensorOps(stream, element_type, algo_desc)); in GetCudnnConvolutionBackwardDataAlgorithm()
2803 UseTensorOps(stream, element_type, algo_desc)); in GetCudnnConvolutionBackwardDataAlgorithm()
2825 UseTensorOps(stream, element_type, algo_desc)); in GetCudnnConvolutionBackwardFilterAlgorithm()
2864 UseTensorOps(stream, element_type, algo_desc)); in GetCudnnConvolutionBackwardFilterAlgorithm()
3057 UseTensorOps(stream, element_type, algorithm_desc)); in DoConvolve()
|