Searched defs:algorithm_desc (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2214 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionForwardWorkspace()
2257 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionBackwardDataWorkspace()
2302 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionBackwardFilterWorkspace()
2633 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
2690 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 611 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveWithAlgorithm() local
653 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveWithAlgorithm() local
695 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveWithAlgorithm() local
840 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveBackwardDataWithAlgorithm() local
885 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveBackwardDataWithAlgorithm() local
930 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveBackwardDataWithAlgorithm() local
975 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveBackwardFilterWithAlgorithm() local
1020 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveBackwardFilterWithAlgorithm() local
1065 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveBackwardFilterWithAlgorithm() local
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 2599 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
2816 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()
|