| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1252 DeviceMemory<uint8>* scratch_memory) { in PrepareForConvolution() argument
1257 scratch_allocator, algorithm_desc, scratch_memory); in PrepareForConvolution()
1301 AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory,
1313 DeviceMemory<uint8>* scratch_memory, in DoConvolve() argument
1320 algorithm_desc, *scratch_memory, output_profile_result), in DoConvolve()
1416 DeviceMemory<uint8>* scratch_memory, in DoConvolveBackwardData() argument
1424 algorithm_desc, *scratch_memory, output_profile_result), in DoConvolveBackwardData()
1463 DeviceMemory<uint8>* scratch_memory, in DoConvolveBackwardFilter() argument
1471 algorithm_desc, *scratch_memory, output_profile_result), in DoConvolveBackwardFilter()
2381 DeviceMemory<uint8>* scratch_memory, in PrepareForCtcLoss() argument
[all …]
|
| D | stream.h | 336 DeviceMemory<uint8> scratch_memory; in ConvolveWithAlgorithm() local
343 &algorithm_desc, &scratch_memory)); in ConvolveWithAlgorithm()
348 *output, convolution_descriptor, algorithm_desc, scratch_memory, in ConvolveWithAlgorithm()
447 DeviceMemory<uint8> scratch_memory; in ConvolveBackwardDataWithAlgorithm() local
455 &scratch_memory)); in ConvolveBackwardDataWithAlgorithm()
462 algorithm_desc, scratch_memory, output_profile_result); in ConvolveBackwardDataWithAlgorithm()
479 DeviceMemory<uint8> scratch_memory; in ConvolveBackwardFilterWithAlgorithm() local
487 &scratch_memory)); in ConvolveBackwardFilterWithAlgorithm()
494 algorithm_desc, scratch_memory, output_profile_result); in ConvolveBackwardFilterWithAlgorithm()
|
| D | dnn.cc | 630 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id) { in DoCtcLoss() argument
|
| D | stream.cc | 4783 DeviceMemory<uint8> scratch_memory; in ThenCtcLoss() local
4789 &scratch_memory, &ctc_loss_algo_id) in ThenCtcLoss()
4795 &scratch_memory, ctc_loss_algo_id); in ThenCtcLoss()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 2397 ScratchAllocator* scratch_allocator, DeviceMemory<uint8>* scratch_memory, in DoPrepareForCtcLoss() argument
2425 *scratch_memory = DeviceMemory<uint8>(); in DoPrepareForCtcLoss()
2436 *scratch_memory = scratch_or.ValueOrDie(); in DoPrepareForCtcLoss()
2461 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id) { in DoCtcLossImpl() argument
2475 scratch_memory.opaque(), scratch_memory.size()); in DoCtcLossImpl()
2491 DeviceMemoryBase grads_data, DeviceMemory<uint8> scratch_memory, in DoCtcLoss() argument
2511 scratch_memory, ctc_loss_algo_id); in DoCtcLoss()
2852 DeviceMemory<uint8>* scratch_memory) { in DoPrepareForConvolution() argument
2874 *scratch_memory = allocated.ValueOrDie(); in DoPrepareForConvolution()
2946 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve() argument
[all …]
|
| D | rocm_dnn.h | 315 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory,
683 DeviceMemory<uint8> scratch_memory,
843 DeviceMemory<uint8>* scratch_memory) override;
852 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id);
861 ScratchAllocator* scratch_allocator, DeviceMemory<uint8>* scratch_memory,
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2037 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id) { in DoCtcLossImpl() argument
2058 /*workspace=*/scratch_memory.opaque(), in DoCtcLossImpl()
2059 /*workSpaceSizeInBytes=*/scratch_memory.size())); in DoCtcLossImpl()
2992 DeviceMemory<uint8>* scratch_memory) { in DoPrepareForConvolution() argument
3011 output_nd, scratch_allocator, scratch_memory)); in DoPrepareForConvolution()
3019 output_nd, scratch_allocator, scratch_memory)); in DoPrepareForConvolution()
3027 output_nd, scratch_allocator, scratch_memory)); in DoPrepareForConvolution()
3046 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve() argument
3116 /*workSpace=*/scratch_memory.opaque(), in DoConvolve()
3117 /*workSpaceSizeInBytes=*/scratch_memory.size(), /*beta=*/beta, in DoConvolve()
[all …]
|
| D | cuda_dnn.h | 277 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory,
571 DeviceMemory<uint8> scratch_memory,
691 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id);
704 DeviceMemory<uint8>* scratch_memory) override;
713 ScratchAllocator* scratch_allocator, DeviceMemory<uint8>* scratch_memory,
|