Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_conv_algorithm_picker.cc431 se::RedzoneAllocator* input_output_allocator, se::Stream* stream, in AutotuneOneConvRunner()
665 se::RedzoneAllocator input_output_allocator( in PickBestAlgorithmNoCacheCuda() local
802 ScratchAllocator input_output_allocator(device_ordinal, allocator); in PickBestAlgorithmNoCacheRocm() local