| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1251 ScratchAllocator* scratch_allocator, AlgorithmDesc* algorithm_desc, in PrepareForConvolution() argument
1257 scratch_allocator, algorithm_desc, scratch_memory); in PrepareForConvolution()
1301 AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory,
1312 const dnn::AlgorithmDesc& algorithm_desc, in DoConvolve() argument
1320 algorithm_desc, *scratch_memory, output_profile_result), in DoConvolve()
1415 const dnn::AlgorithmDesc& algorithm_desc, in DoConvolveBackwardData() argument
1424 algorithm_desc, *scratch_memory, output_profile_result), in DoConvolveBackwardData()
1462 const dnn::AlgorithmDesc& algorithm_desc, in DoConvolveBackwardFilter() argument
1471 algorithm_desc, *scratch_memory, output_profile_result), in DoConvolveBackwardFilter()
2686 ScratchAllocator* scratch_allocator, AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution() argument
[all …]
|
| D | stream.h | 337 dnn::AlgorithmDesc algorithm_desc; in ConvolveWithAlgorithm() local
343 &algorithm_desc, &scratch_memory)); in ConvolveWithAlgorithm()
348 *output, convolution_descriptor, algorithm_desc, scratch_memory, in ConvolveWithAlgorithm()
448 dnn::AlgorithmDesc algorithm_desc; in ConvolveBackwardDataWithAlgorithm() local
454 algorithm_config, scratch_allocator, &algorithm_desc, in ConvolveBackwardDataWithAlgorithm()
462 algorithm_desc, scratch_memory, output_profile_result); in ConvolveBackwardDataWithAlgorithm()
480 dnn::AlgorithmDesc algorithm_desc; in ConvolveBackwardFilterWithAlgorithm() local
486 algorithm_config, scratch_allocator, &algorithm_desc, in ConvolveBackwardFilterWithAlgorithm()
494 algorithm_desc, scratch_memory, output_profile_result); in ConvolveBackwardFilterWithAlgorithm()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2539 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionForwardWorkspace() argument
2541 if (IsTensorMathOpSet(conv) != algorithm_desc.tensor_ops_enabled()) { in AllocateCudnnConvolutionForwardWorkspace()
2553 /*yDesc=*/output_nd.handle(), /*algo=*/ToConvForwardAlgo(algorithm_desc), in AllocateCudnnConvolutionForwardWorkspace()
2583 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionBackwardDataWorkspace() argument
2585 if (IsTensorMathOpSet(conv) != algorithm_desc.tensor_ops_enabled()) { in AllocateCudnnConvolutionBackwardDataWorkspace()
2599 /*algo=*/ToConvBackwardDataAlgo(algorithm_desc), in AllocateCudnnConvolutionBackwardDataWorkspace()
2629 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionBackwardFilterWorkspace() argument
2631 if (IsTensorMathOpSet(conv) != algorithm_desc.tensor_ops_enabled()) { in AllocateCudnnConvolutionBackwardFilterWorkspace()
2645 /*algo=*/ToConvBackwardFilterAlgo(algorithm_desc), in AllocateCudnnConvolutionBackwardFilterWorkspace()
2991 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution() argument
[all …]
|
| D | cuda_dnn.h | 277 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory,
703 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 2851 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution() argument
2859 *algorithm_desc = *input_algo_desc; in DoPrepareForConvolution()
2946 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve() argument
2989 static_cast<uint64_t>(algorithm_desc.algo_id())); in DoConvolve()
2994 static_cast<miopenConvFwdAlgorithm_t>(algorithm_desc.algo_id()), in DoConvolve()
3016 static_cast<uint64_t>(algorithm_desc.algo_id())); in DoConvolve()
3021 static_cast<miopenConvBwdDataAlgorithm_t>(algorithm_desc.algo_id()), in DoConvolve()
3042 static_cast<uint64_t>(algorithm_desc.algo_id())); in DoConvolve()
3048 algorithm_desc.algo_id()), in DoConvolve()
3065 dnn::AlgorithmDesc algotype(algorithm_desc.algo_id(), false); in DoConvolve()
|
| D | rocm_dnn.h | 315 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory,
842 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc,
|