Searched refs:output_rows_host (Results 1 – 1 of 1) sorted by relevance
250 ScratchSpace<Index> output_rows_host(context, 1, /* on_host */ true); in ComputeAsync() local256 ->ThenMemcpy(output_rows_host.mutable_data(), output_rows_device, in ComputeAsync()264 auto create_and_check_output = [context, output_rows_host, &input, in ComputeAsync()271 Index output_rows = *output_rows_host.data(); in ComputeAsync()