Home
last modified time | relevance | path

Searched defs:algorithm_desc (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dstream.h348 dnn::AlgorithmDesc algorithm_desc; in ConvolveWithAlgorithm() local
394 const dnn::AlgorithmDesc &algorithm_desc, dnn::ConvolutionKind kind, in ConvolveRunnerFromDesc()
411 const dnn::AlgorithmDesc &algorithm_desc, dnn::ConvolutionKind kind, in FusedConvolveRunnerFromDesc()
Ddnn.cc133 Stream* stream, const dnn::AlgorithmDesc& algorithm_desc, in ConvolveRunnerFromDesc()
159 Stream* stream, const dnn::AlgorithmDesc& algorithm_desc, in FusedConvolveRunnerFromDesc()
Ddnn.h1330 ScratchAllocator* scratch_allocator, AlgorithmDesc* algorithm_desc, in PrepareForConvolution()
2626 ScratchAllocator* scratch_allocator, AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc2891 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionForwardWorkspace()
2940 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionBackwardDataWorkspace()
2990 const dnn::AlgorithmDesc& algorithm_desc, in AllocateCudnnConvolutionBackwardFilterWorkspace()
3964 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
4264 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()
4817 Stream* stream, const dnn::AlgorithmDesc& algorithm_desc, in ConvolveRunnerFromDesc()
5063 Stream* stream, const dnn::AlgorithmDesc& algorithm_desc, in FusedConvolveRunnerFromDesc()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2911 ScratchAllocator* scratch_allocator, dnn::AlgorithmDesc* algorithm_desc, in DoPrepareForConvolution()
3119 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()
3187 Stream* stream, const dnn::AlgorithmDesc& algorithm_desc, in ConvolveRunnerFromDesc()