| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | copy_thunk.cc | 37 params.stream->ThenMemcpy(&destination_data, source_address_, mem_size_); in ExecuteOnStream()
56 params.stream->ThenMemcpy(&destination_data, source_data, mem_size_); in ExecuteOnStream()
|
| D | conditional_thunk.cc | 81 stream.ThenMemcpy(&pred, branch_index_address, sizeof(bool)); in ExecuteOnStream()
83 stream.ThenMemcpy(&branch_index, branch_index_address, sizeof(int32)); in ExecuteOnStream()
|
| D | buffer_comparator.cc | 631 stream->ThenMemcpy(&result, *out_param, sizeof(result)); in DeviceCompare()
645 stream->ThenMemcpy(host_lhs.data(), lhs, lhs.size()); in HostCompare()
646 stream->ThenMemcpy(host_rhs.data(), rhs, rhs.size()); in HostCompare()
|
| D | infeed_thunk.cc | 58 stream.ThenMemcpy(&dest_address, *buffer.device_memory(), buffer.length()); in ExecuteOnStream()
|
| D | while_thunk.cc | 72 stream.ThenMemcpy(&condition_result, condition_result_data, sizeof(bool)); in ExecuteOnStream()
|
| D | outfeed_thunk.cc | 96 .ThenMemcpy(buffer->destination()->untyped_data(), data_address, in ExecuteOnStream()
|
| D | buffer_comparator_test.cc | 50 stream.ThenMemcpy(lhs_buffer.ptr(), lhs.data(), lhs_buffer->size()); in CompareEqualBuffers()
51 stream.ThenMemcpy(rhs_buffer.ptr(), rhs.data(), rhs_buffer->size()); in CompareEqualBuffers()
|
| D | thunk.h | 135 stream->ThenMemcpy(&dest, buf.get(), count * sizeof(T)); in SafeH2DMemcpy()
|
| D | kernel_thunk.cc | 75 CHECK(stream->ThenMemcpy(host_buffer.get(), buf, buf.size()).ok()); in PrintBufferContents()
|
| D | gpu_transfer_manager.cc | 109 stream->ThenMemcpy(buffer.device_memory(), source, size); in TransferBufferToInfeedInternal()
|
| D | gemm_algorithm_picker.cc | 175 stream->ThenMemcpy(&reference_result_buffer, output_buffer, in DoUncachedGemmAutotune()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | redzone_allocator.cc | 109 stream_->ThenMemcpy(&rhs_redzone_slop, &pattern32, rhs_slop); in AllocateBytes()
186 TF_RETURN_IF_ERROR(stream->ThenMemcpy(redzone_data.get(), redzone, size) in CheckRedzoneHost()
242 stream->ThenMemcpy(&redzone, redzone_array.data(), redzone.size()); in ReinitializeRedzone()
278 stream->ThenMemcpy(&result, out_param, sizeof(result)); in CheckRedzonesForBuffer()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 161 send_device_to_host_stream->ThenMemcpy(buf, gpu_src_ptr, total_bytes); in SetProtoFromGPU()
231 send_device_to_device_stream->ThenMemcpy(&gpu_dst_ptr, gpu_src_ptr, in DeviceToDeviceCopy()
283 send_device_to_host_stream->ThenMemcpy(dst_ptr, gpu_src_ptr, total_bytes); in CopyGPUTensorToCPU()
331 recv_host_to_device_stream->ThenMemcpy(&gpu_dst_ptr, src_ptr, total_bytes); in CopyCPUTensorToGPU()
443 send_stream->ThenMemcpy(&gpu_dst_ptr, gpu_src_ptr, total_bytes); in CopyGPUTensorToSameGPU()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | redzone_allocator_test.cc | 77 TF_ASSERT_OK(stream.ThenMemcpy(host_buf.data(), redzone, kRedzoneSize) in TEST()
106 stream.ThenMemcpy(&old_redzone_value, redzone_at_offset, 1) in TEST()
|
| D | memcpy_test.cc | 37 stream.ThenMemcpy(&h_ptr, d_mem, d_mem.size()); in TEST()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | mlir_gpu_test_base.cc | 126 stream->ThenMemcpy(&memory, static_cast<void*>(arguments[i].data()), in RunMlirModuleWithHostBuffers()
137 stream->ThenMemcpy(static_cast<void*>(host_outputs.back().data()), in RunMlirModuleWithHostBuffers()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_device_array.h | 88 stream->ThenMemcpy(&output_values_base, in Finalize()
|
| D | debug_ops.h | 742 stream->ThenMemcpy(&output_tensor_ptr, &tensor_id, sizeof(Tout));
769 stream->ThenMemcpy(&output_tensor_ptr, &static_output, 2 * sizeof(Tout));
801 stream->ThenMemcpy(&output_tensor_ptr, &static_output, 5 * sizeof(Tout));
844 stream->ThenMemcpy(&output_tensor_ptr, &static_output, sizeof(Tout) * 10);
|
| D | check_numerics_op.cc | 250 ->ThenMemcpy(abnormal_detected_host.flat<int>().data(), in ComputeAsync()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | generic_transfer_manager.cc | 76 stream->ThenMemcpy( in TransferLiteralFromDevice()
|
| D | transfer_manager.cc | 370 stream->ThenMemcpy(destination, source, size); in TransferBufferFromDevice()
383 stream->ThenMemcpy(destination, source, size); in TransferBufferToDevice()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr_test.cc | 388 gpu_helper_->h2d_stream()->ThenMemcpy( in DoAddChain()
392 gpu_helper_->h2d_stream()->ThenMemcpy( in DoAddChain()
422 gpu_helper_->d2h_stream()->ThenMemcpy(DMAHelper::base(&host_outputs_[0]), in DoAddChain()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager_test.cc | 157 stream->ThenMemcpy(&in_gpu_mem, in_cpu.flat<Scalar>().data(), in MakeReductionTestCase()
198 stream->ThenMemcpy(&in_gpu_mem, in_cpu.flat<Scalar>().data(), in MakeGatherTestCase()
235 stream->ThenMemcpy(&in_gpu_mem, in_cpu.flat<Scalar>().data(), in MakeBroadcastTestCase()
270 stream->ThenMemcpy(out_cpu.flat<Scalar>().data(), out_gpu_mem, in VerifyResults()
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_test.cc | 728 Stream& stream_ref = stream.ThenMemcpy(&dst_data, device_src, size); in TEST_F()
749 stream.ThenMemcpy(&device_dst, &src_data, size); in TEST_F()
770 stream.ThenMemcpy(&device_dst, device_src, size); in TEST_F()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_tensor_to_csr_sparse_matrix_op.cc | 165 ->ThenMemcpy(nnz_per_batch_host.mutable_data() /*host_dst*/, in ComputeAsync()
|