Searched refs:is_host2device (Results 1 – 2 of 2) sorted by relevance
85 bool is_host2device, bool sync) { in SyncMemInHostAndDevice() argument90 …ostAndDevice(host_tensor.MutableData(), device_tensor_name, host_tensor.DataSize(), is_host2device, in SyncMemInHostAndDevice()95 bool is_host2device, bool sync) { in SyncMemInHostAndDevice() argument102 current_cuda_tensor.is_valid_mem = is_host2device ? current_cuda_tensor.is_valid_mem : true; in SyncMemInHostAndDevice()103 if (is_host2device && current_cuda_tensor.is_valid_mem) { in SyncMemInHostAndDevice()113 void *src_ptr = is_host2device ? host_data : device_ptr; in SyncMemInHostAndDevice()114 void *dst_ptr = is_host2device ? device_ptr : host_data; in SyncMemInHostAndDevice()115 cudaMemcpyKind kind = is_host2device ? cudaMemcpyHostToDevice : cudaMemcpyDeviceToHost; in SyncMemInHostAndDevice()
45 bool is_host2device, bool sync = true);48 bool is_host2device, bool sync = true);