Searched defs:algorithm_desc (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 669 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveWithAlgorithm() local
711 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveWithAlgorithm() local
753 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveWithAlgorithm() local
795 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveWithAlgorithm() local
837 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveWithAlgorithm() local
982 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveBackwardDataWithAlgorithm() local
1027 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveBackwardDataWithAlgorithm() local
1072 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveBackwardDataWithAlgorithm() local
1117 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveBackwardFilterWithAlgorithm() local
1162 dnn::AlgorithmDesc algorithm_desc; in ThenConvolveBackwardFilterWithAlgorithm() local
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2466 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionForwardWorkspace()
2509 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionBackwardDataWorkspace()
2554 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionBackwardFilterWorkspace()
2901 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
2959 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 2644 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
2809 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()
|