Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dstream.h341 DeviceMemory<uint8> scratch_memory; in ConvolveWithAlgorithm() local
493 DeviceMemory<uint8> scratch_memory; in ConvolveBackwardDataWithAlgorithm() local
525 DeviceMemory<uint8> scratch_memory; in ConvolveBackwardFilterWithAlgorithm() local
Ddnn.h1283 DeviceMemory<uint8>* scratch_memory) { in PrepareForConvolution()
2289 DeviceMemory<uint8>* scratch_memory, in PrepareForCtcLoss()
2337 DeviceMemory<uint8>* scratch_memory, int ctc_loss_algo_id) { in DoCtcLoss()
2595 DeviceMemory<uint8>* scratch_memory) { in DoPrepareForConvolution()
2608 ScratchAllocator* scratch_allocator, DeviceMemory<uint8>* scratch_memory, in DoPrepareForCtcLoss()
Ddnn.cc776 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id) { in DoCtcLoss()
Dstream.cc4235 DeviceMemory<uint8> scratch_memory; in ThenCtcLoss() local
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc2320 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id) { in DoCtcLossImpl()
3872 DeviceMemory<uint8>* scratch_memory) { in DoPrepareForConvolution()
3926 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()
4107 DeviceMemory<uint8> scratch_memory; in DoConvolveWithExecutionPlan() local
5069 DeviceMemory<uint8> scratch_memory; in DoFusedConvolveWithExecutionPlan() local
5153 ScratchAllocator* scratch_allocator, DeviceMemory<uint8>* scratch_memory, in DoPrepareForCtcLoss()
5223 DeviceMemoryBase grads_data, DeviceMemory<uint8> scratch_memory, in DoCtcLoss()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2448 ScratchAllocator* scratch_allocator, DeviceMemory<uint8>* scratch_memory, in DoPrepareForCtcLoss()
2512 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id) { in DoCtcLossImpl()
2542 DeviceMemoryBase grads_data, DeviceMemory<uint8> scratch_memory, in DoCtcLoss()
2909 DeviceMemory<uint8>* scratch_memory) { in DoPrepareForConvolution()
2956 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()
3390 DeviceMemory<uint8> scratch_memory; in GetMIOpenConvolveAlgorithmsFindMode() local