Home
last modified time | relevance | path

Searched refs:reference_result_buffer (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgemm_algorithm_picker.cc94 TF_ASSIGN_OR_RETURN(se::DeviceMemoryBase reference_result_buffer, in DoUncachedGemmAutotune()
174 CHECK(reference_result_buffer.size() == output_buffer.size()); in DoUncachedGemmAutotune()
175 stream->ThenMemcpy(&reference_result_buffer, output_buffer, in DoUncachedGemmAutotune()
182 reference_result_buffer)); in DoUncachedGemmAutotune()
Dgpu_conv_algorithm_picker.cc425 se::DeviceMemoryBase reference_result_buffer; in PickBestAlgorithmNoCacheCuda() local
533 stream, reference_result_buffer, result_buffer); in PickBestAlgorithmNoCacheCuda()
567 reference_result_buffer, in PickBestAlgorithmNoCacheCuda()
569 stream->ThenMemcpy(&reference_result_buffer, result_buffer, in PickBestAlgorithmNoCacheCuda()