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/
Dgpu_conv_algorithm_picker.cc390 se::RedzoneAllocator input_output_allocator( in PickBestAlgorithmNoCacheCuda() local
648 ScratchAllocator input_output_allocator(device_ordinal, allocator); in PickBestAlgorithmNoCacheRocm() local
Dgemm_algorithm_picker.cc261 se::RedzoneAllocator input_output_allocator( in RunOnInstruction() local