Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_conv_runner.cc75 DeviceMemoryBase scratch_memory) { in RunGpuConvUnfused()
121 DeviceMemoryBase scratch_memory) { in RunGpuConvForwardActivation()
193 DeviceMemoryBase scratch_memory) { in RunGpuConvInternalImpl()
218 DeviceMemoryBase scratch_memory) { in RunGpuConvInternalImpl()
237 se::DeviceMemoryBase scratch_memory, in RunGpuConvImpl()
520 se::DeviceMemoryBase scratch_memory, se::Stream* stream, in RunGpuConv()
H A Dgpu_conv_algorithm_picker.cc511 se::DeviceMemoryBase scratch_memory = scratch_or.ValueOrDie(); in AutotuneOneConvRunner() local
/aosp_15_r20/external/webrtc/modules/audio_processing/aec3/
H A Dmatched_filter.cc164 rtc::ArrayView<float> scratch_memory) { in MatchedFilterCoreWithAccumulatedError_NEON()
246 rtc::ArrayView<float> scratch_memory) { in MatchedFilterCore_NEON()
363 rtc::ArrayView<float> scratch_memory) { in MatchedFilterCore_AccumulatedError_SSE2()
460 rtc::ArrayView<float> scratch_memory) { in MatchedFilterCore_SSE2()
H A Dmatched_filter_avx2.cc41 rtc::ArrayView<float> scratch_memory) { in MatchedFilterCore_AccumulatedError_AVX2()
149 rtc::ArrayView<float> scratch_memory) { in MatchedFilterCore_AVX2()
H A Dmatched_filter_unittest.cc70 std::vector<float> scratch_memory(512); in TEST_P() local
132 std::vector<float> scratch_memory(512); in TEST_P() local
188 std::vector<float> scratch_memory(512); in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h1331 DeviceMemory<uint8>* scratch_memory) { in PrepareForConvolution()
2315 DeviceMemory<uint8>* scratch_memory, in PrepareForCtcLoss()
2363 DeviceMemory<uint8>* scratch_memory, int ctc_loss_algo_id) { in DoCtcLoss()
2627 DeviceMemory<uint8>* scratch_memory) { in DoPrepareForConvolution()
2640 ScratchAllocator* scratch_allocator, DeviceMemory<uint8>* scratch_memory, in DoPrepareForCtcLoss()
H A Dstream.h347 DeviceMemory<uint8> scratch_memory; in ConvolveWithAlgorithm() local
H A Ddnn.cc882 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id) { in DoCtcLoss()
H A Dstream.cc2284 DeviceMemory<uint8> scratch_memory; in ThenCtcLoss() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_ops_gpu.h140 se::DeviceMemoryBase scratch_memory; in AllocateScratchOrFallback() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc2380 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id) { in DoCtcLossImpl()
3965 DeviceMemory<uint8>* scratch_memory) { in DoPrepareForConvolution()
4081 DeviceMemoryBase scratch_memory, in operator ()()
4264 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()
4451 DeviceMemoryBase scratch_memory, in operator ()()
4919 DeviceMemoryBase scratch_memory, in operator ()()
5843 ScratchAllocator* scratch_allocator, DeviceMemory<uint8>* scratch_memory, in DoPrepareForCtcLoss()
5913 DeviceMemoryBase grads_data, DeviceMemory<uint8> scratch_memory, in DoCtcLoss()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc2451 ScratchAllocator* scratch_allocator, DeviceMemory<uint8>* scratch_memory, in DoPrepareForCtcLoss()
2515 DeviceMemory<uint8> scratch_memory, int ctc_loss_algo_id) { in DoCtcLossImpl()
2545 DeviceMemoryBase grads_data, DeviceMemory<uint8> scratch_memory, in DoCtcLoss()
2912 DeviceMemory<uint8>* scratch_memory) { in DoPrepareForConvolution()
2982 DeviceMemoryBase scratch_memory, in operator ()()
3119 dnn::AlgorithmDesc algorithm_desc, DeviceMemory<uint8> scratch_memory, in DoConvolve()
3511 DeviceMemory<uint8> scratch_memory; in GetMIOpenConvolveAlgorithmsFindMode() local