Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_algorithm_picker.cc70 se::RedzoneAllocator input_output_allocator( in DoUncachedGemmAutotune() local
Dgpu_conv_algorithm_picker.cc403 se::RedzoneAllocator input_output_allocator( in PickBestAlgorithmNoCacheCuda() local
663 ScratchAllocator input_output_allocator(device_ordinal, allocator); in PickBestAlgorithmNoCacheRocm() local