Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1198 DeviceMemory<uint8>* scratch_memory) {
1203 scratch_allocator, algorithm_desc, scratch_memory);
1246 AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory,
1258 DeviceMemory<uint8>* scratch_memory,
1264 convolution_descriptor, algorithm_desc, *scratch_memory,
1350 DeviceMemory<uint8>* scratch_memory,
1357 convolution_descriptor, algorithm_desc, *scratch_memory,
1397 DeviceMemory<uint8>* scratch_memory,
1404 convolution_descriptor, algorithm_desc, *scratch_memory,
2552 DeviceMemory<uint8>* scratch_memory) {
[all …]
Dstream.cc610 DeviceMemory<uint8> scratch_memory; in ThenConvolveWithAlgorithm() local
617 scratch_allocator, &algorithm_desc, &scratch_memory) in ThenConvolveWithAlgorithm()
623 &scratch_memory, output_profile_result); in ThenConvolveWithAlgorithm()
652 DeviceMemory<uint8> scratch_memory; in ThenConvolveWithAlgorithm() local
659 scratch_allocator, &algorithm_desc, &scratch_memory) in ThenConvolveWithAlgorithm()
665 &scratch_memory, output_profile_result); in ThenConvolveWithAlgorithm()
694 DeviceMemory<uint8> scratch_memory; in ThenConvolveWithAlgorithm() local
701 scratch_allocator, &algorithm_desc, &scratch_memory) in ThenConvolveWithAlgorithm()
707 &scratch_memory, output_profile_result); in ThenConvolveWithAlgorithm()
839 DeviceMemory<uint8> scratch_memory; in ThenConvolveBackwardDataWithAlgorithm() local
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc2634 DeviceMemory<uint8>* scratch_memory) { in DoPrepareForConvolution() argument
2656 output_nd, scratch_allocator, scratch_memory)); in DoPrepareForConvolution()
2664 output_nd, scratch_allocator, scratch_memory)); in DoPrepareForConvolution()
2672 output_nd, scratch_allocator, scratch_memory)); in DoPrepareForConvolution()
2690 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve() argument
2785 stream->ThenMemZero(&scratch_memory, scratch_memory.size()); in DoConvolve()
2862 /*workSpace=*/scratch_memory.opaque(), in DoConvolve()
2863 /*workSpaceSizeInBytes=*/scratch_memory.size(), /*beta=*/beta, in DoConvolve()
2878 /*workSpace=*/scratch_memory.opaque(), in DoConvolve()
2879 /*workSpaceSizeInBytes=*/scratch_memory.size(), in DoConvolve()
[all …]
Dcuda_dnn.h270 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory,
661 DeviceMemory<uint8>* scratch_memory) override;
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2600 DeviceMemory<uint8>* scratch_memory) { in DoPrepareForConvolution() argument
2755 *scratch_memory = allocated.ValueOrDie(); in DoPrepareForConvolution()
2816 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve() argument
2862 /*workSpace=*/scratch_memory.opaque(), in DoConvolve()
2863 /*workSpaceSizeInBytes=*/scratch_memory.size()); in DoConvolve()
2888 /*workSpace=*/scratch_memory.opaque(), in DoConvolve()
2889 /*workSpaceSizeInBytes=*/scratch_memory.size()); in DoConvolve()
2915 /*workSpace=*/scratch_memory.opaque(), in DoConvolve()
2916 /*workSpaceSizeInBytes=*/scratch_memory.size()); in DoConvolve()
2934 output_profile_result->set_scratch_size(scratch_memory.size()); in DoConvolve()
Drocm_dnn.h261 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory,
768 DeviceMemory<uint8>* scratch_memory) override;