Searched refs:input_output_allocator (Results 1 – 2 of 2) sorted by relevance
403 se::RedzoneAllocator input_output_allocator( in PickBestAlgorithmNoCacheCuda() local408 input_output_allocator.AllocateBytes( in PickBestAlgorithmNoCacheCuda()414 input_output_allocator.AllocateBytes( in PickBestAlgorithmNoCacheCuda()502 CheckRedzones(input_output_allocator, stream, in PickBestAlgorithmNoCacheCuda()568 input_output_allocator.AllocateBytes(result_buffer.size())); in PickBestAlgorithmNoCacheCuda()663 ScratchAllocator input_output_allocator(device_ordinal, allocator); in PickBestAlgorithmNoCacheRocm() local677 input_output_allocator.AllocateBytes( in PickBestAlgorithmNoCacheRocm()685 input_output_allocator.AllocateBytes( in PickBestAlgorithmNoCacheRocm()
70 se::RedzoneAllocator input_output_allocator( in DoUncachedGemmAutotune() local80 input_output_allocator.AllocateBytes( in DoUncachedGemmAutotune()162 input_output_allocator.CheckRedzones()); in DoUncachedGemmAutotune()