Searched refs:input_output_allocator (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_conv_algorithm_picker.cc | 431 se::RedzoneAllocator* input_output_allocator, se::Stream* stream, in AutotuneOneConvRunner() argument
556 CheckRedzones(*input_output_allocator, stream, in AutotuneOneConvRunner()
632 input_output_allocator->AllocateBytes(result_buffer.size())); in AutotuneOneConvRunner()
665 se::RedzoneAllocator input_output_allocator( in PickBestAlgorithmNoCacheCuda() local
673 input_output_allocator.AllocateBytes( in PickBestAlgorithmNoCacheCuda()
679 input_output_allocator.AllocateBytes( in PickBestAlgorithmNoCacheCuda()
719 config, instr, allocator, &input_output_allocator, in PickBestAlgorithmNoCacheCuda()
739 config, instr, allocator, &input_output_allocator, in PickBestAlgorithmNoCacheCuda()
802 ScratchAllocator input_output_allocator(device_ordinal, allocator); in PickBestAlgorithmNoCacheRocm() local
816 input_output_allocator.AllocateBytes( in PickBestAlgorithmNoCacheRocm()
[all …]
|
| H A D | gpu_conv_algorithm_picker.h | 75 se::RedzoneAllocator* input_output_allocator, se::Stream* stream,
|