| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2214 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionForwardWorkspace() argument
2219 conv.set_use_tensor_op_math(algorithm_desc.tensor_ops_enabled()); in AllocateCudnnConvolutionForwardWorkspace()
2227 /*yDesc=*/output_nd.handle(), /*algo=*/ToConvForwardAlgo(algorithm_desc), in AllocateCudnnConvolutionForwardWorkspace()
2257 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionBackwardDataWorkspace() argument
2262 conv.set_use_tensor_op_math(algorithm_desc.tensor_ops_enabled()); in AllocateCudnnConvolutionBackwardDataWorkspace()
2272 /*algo=*/ToConvBackwardDataAlgo(algorithm_desc), in AllocateCudnnConvolutionBackwardDataWorkspace()
2302 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionBackwardFilterWorkspace() argument
2307 conv.set_use_tensor_op_math(algorithm_desc.tensor_ops_enabled()); in AllocateCudnnConvolutionBackwardFilterWorkspace()
2317 /*algo=*/ToConvBackwardFilterAlgo(algorithm_desc), in AllocateCudnnConvolutionBackwardFilterWorkspace()
2633 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution() argument
[all …]
|
| D | cuda_dnn.h | 270 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory,
660 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc,
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1197 ScratchAllocator* scratch_allocator, AlgorithmDesc* algorithm_desc,
1203 scratch_allocator, algorithm_desc, scratch_memory);
1246 AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory,
1257 const dnn::AlgorithmDesc& algorithm_desc,
1264 convolution_descriptor, algorithm_desc, *scratch_memory,
1349 const dnn::AlgorithmDesc& algorithm_desc,
1357 convolution_descriptor, algorithm_desc, *scratch_memory,
1396 const dnn::AlgorithmDesc& algorithm_desc,
1404 convolution_descriptor, algorithm_desc, *scratch_memory,
2551 ScratchAllocator* scratch_allocator, AlgorithmDesc* algorithm_desc,
[all …]
|
| D | stream.cc | 611 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveWithAlgorithm() local
617 scratch_allocator, &algorithm_desc, &scratch_memory) in ThenConvolveWithAlgorithm()
622 convolution_descriptor, output_descriptor, output, algorithm_desc, in ThenConvolveWithAlgorithm()
653 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveWithAlgorithm() local
659 scratch_allocator, &algorithm_desc, &scratch_memory) in ThenConvolveWithAlgorithm()
664 convolution_descriptor, output_descriptor, output, algorithm_desc, in ThenConvolveWithAlgorithm()
695 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveWithAlgorithm() local
701 scratch_allocator, &algorithm_desc, &scratch_memory) in ThenConvolveWithAlgorithm()
706 convolution_descriptor, output_descriptor, output, algorithm_desc, in ThenConvolveWithAlgorithm()
840 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveBackwardDataWithAlgorithm() local
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 2599 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution() argument
2681 *algorithm_desc = dnn::AlgorithmDesc(preference.fwd_algo, false); in DoPrepareForConvolution()
2699 *algorithm_desc = dnn::AlgorithmDesc(preference.bwd_data_algo, false); in DoPrepareForConvolution()
2717 *algorithm_desc = in DoPrepareForConvolution()
2732 *algorithm_desc = *algo_desc; in DoPrepareForConvolution()
2816 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve() argument
2859 static_cast<miopenConvFwdAlgorithm_t>(algorithm_desc.algo_id()), in DoConvolve()
2884 static_cast<miopenConvBwdDataAlgorithm_t>(algorithm_desc.algo_id()), in DoConvolve()
2911 algorithm_desc.algo_id()), in DoConvolve()
2930 dnn::AlgorithmDesc algotype(algorithm_desc.algo_id(), false); in DoConvolve()
|
| D | rocm_dnn.h | 261 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory,
767 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc,
|