Searched refs:initialize_buffer (Results 1 – 1 of 1) sorted by relevance
394 const auto initialize_buffer = [init_conv_data, &stream, &rng_state]( in PickBestAlgorithmNoCacheCuda() local410 initialize_buffer(buffer, operand->shape()); in PickBestAlgorithmNoCacheCuda()416 initialize_buffer(result_buffer, result_shape); in PickBestAlgorithmNoCacheCuda()664 const auto initialize_buffer = [stream](DeviceMemoryBase buffer) { in PickBestAlgorithmNoCacheRocm() local679 initialize_buffer(buffer); in PickBestAlgorithmNoCacheRocm()687 initialize_buffer(result_buffer); in PickBestAlgorithmNoCacheRocm()