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()
|