Home
last modified time | relevance | path

Searched defs:scratch_memory (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Dstream.h336 DeviceMemory<uint8> scratch_memory; in ConvolveWithAlgorithm() local
447 DeviceMemory<uint8> scratch_memory; in ConvolveBackwardDataWithAlgorithm() local
479 DeviceMemory<uint8> scratch_memory; in ConvolveBackwardFilterWithAlgorithm() local
Ddnn.h1252 DeviceMemory<uint8>* scratch_memory) { in PrepareForConvolution()
1313 DeviceMemory<uint8>* scratch_memory, in DoConvolve()
1416 DeviceMemory<uint8>* scratch_memory, in DoConvolveBackwardData()
1463 DeviceMemory<uint8>* scratch_memory, in DoConvolveBackwardFilter()
2381 DeviceMemory<uint8>* scratch_memory, in PrepareForCtcLoss()
2429 DeviceMemory<uint8>* scratch_memory, int ctc_loss_algo_id) { in DoCtcLoss()
2687 DeviceMemory<uint8>* scratch_memory) { in DoPrepareForConvolution()
2700 ScratchAllocator* scratch_allocator, DeviceMemory<uint8>* scratch_memory, in DoPrepareForCtcLoss()
Ddnn.cc630 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id) { in DoCtcLoss()
Dstream.cc4783 DeviceMemory<uint8> scratch_memory; in ThenCtcLoss() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2397 ScratchAllocator* scratch_allocator, DeviceMemory<uint8>* scratch_memory, in DoPrepareForCtcLoss()
2461 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id) { in DoCtcLossImpl()
2491 DeviceMemoryBase grads_data, DeviceMemory<uint8> scratch_memory, in DoCtcLoss()
2852 DeviceMemory<uint8>* scratch_memory) { in DoPrepareForConvolution()
2946 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()
3396 DeviceMemory<uint8> scratch_memory; in GetMIOpenConvolveAlgorithmsFindMode() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc2037 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id) { in DoCtcLossImpl()
2992 DeviceMemory<uint8>* scratch_memory) { in DoPrepareForConvolution()
3046 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()
3892 ScratchAllocator* scratch_allocator, DeviceMemory<uint8>* scratch_memory, in DoPrepareForCtcLoss()
3962 DeviceMemoryBase grads_data, DeviceMemory<uint8> scratch_memory, in DoCtcLoss()