Searched refs:UseTensorOps (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3034 port::StatusOr<bool> UseTensorOps(Stream* stream, dnn::DataType type, in UseTensorOps() function
3067 UseTensorOps(stream, element_type, algo_desc)); in GetCudnnConvolutionForwardAlgorithm()
3107 UseTensorOps(stream, element_type, algo_desc)); in GetCudnnConvolutionForwardAlgorithm()
3129 UseTensorOps(stream, element_type, algo_desc)); in GetCudnnConvolutionBackwardDataAlgorithm()
3168 UseTensorOps(stream, element_type, algo_desc)); in GetCudnnConvolutionBackwardDataAlgorithm()
3190 UseTensorOps(stream, element_type, algo_desc)); in GetCudnnConvolutionBackwardFilterAlgorithm()
3232 UseTensorOps(stream, element_type, algo_desc)); in GetCudnnConvolutionBackwardFilterAlgorithm()
4088 TF_RETURN_IF_ERROR(UseTensorOps(stream, input_type_, algo).status()); in operator ()()
4280 UseTensorOps(stream, element_type, algorithm_desc)); in DoConvolve()
|