| /external/llvm-project/openmp/libomptarget/src/ |
| D | api.cpp | 127 size_t dst_offset, size_t src_offset, int dst_device, int src_device) { in omp_target_memcpy() argument
130 "src offset %zu, length %zu\n", dst_device, src_device, DPxPTR(dst), in omp_target_memcpy()
138 if (src_device != omp_get_initial_device() && !device_is_ready(src_device)) { in omp_target_memcpy()
152 if (src_device == omp_get_initial_device() && in omp_target_memcpy()
158 } else if (src_device == omp_get_initial_device()) { in omp_target_memcpy()
164 DeviceTy &SrcDev = PM->Devices[src_device]; in omp_target_memcpy()
168 DeviceTy &SrcDev = PM->Devices[src_device]; in omp_target_memcpy()
192 const size_t *src_dimensions, int dst_device, int src_device) { in omp_target_memcpy_rect() argument
197 src_device, DPxPTR(dst), DPxPTR(src), DPxPTR(dst_offsets), in omp_target_memcpy_rect()
217 dst_device, src_device); in omp_target_memcpy_rect()
[all …]
|
| /external/llvm-project/openmp/libomptarget/test/offloading/ |
| D | d2d_memcpy.c | 24 const int src_device = 0; in main() local
28 int *src_ptr = omp_target_alloc(length, src_device); in main()
34 #pragma omp target teams distribute parallel for device(src_device) \ in main()
41 omp_target_memcpy(dst_ptr, src_ptr, length, 0, 0, dst_device, src_device); in main()
62 omp_target_free(src_ptr, src_device); in main()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous.cc | 41 src_device = StringPiece(buf_.data() + (b.src_device.data() - b_base), in operator =()
42 b.src_device.size()); in operator =()
54 string Rendezvous::CreateKey(const string& src_device, uint64 src_incarnation, in CreateKey() argument
66 src_device, ";", strings::Uint64ToHexString(src_incarnation, buf), ";", in CreateKey()
109 out->src_device = StringPiece(parts[0].data(), parts[0].size()); in ParseKey()
|
| D | rendezvous.h | 60 StringPiece src_device; member
172 static std::string CreateKey(const std::string& src_device,
|
| D | local_rendezvous.cc | 121 ->GetCell(string(key.src_device), string(key.dst_device)) in Send()
|
| D | rendezvous_test.cc | 53 EXPECT_EQ(parsed.src_device, "/job:mnist/replica:1/task:2/CPU:0"); in TEST()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_cache_logger.cc | 88 const string& src_device, in RecordRecvTensor() argument
91 RecordDataTransfer(step_id, start_usecs, end_usecs, tensor_name, src_device, in RecordRecvTensor()
98 const string& src_device, in RecordDataTransfer() argument
115 " from ", src_device, " to ", dst_device); in RecordDataTransfer()
|
| D | worker_cache_logger.h | 60 const string& tensor_name, const string& src_device,
66 const string& tensor_name, const string& src_device,
|
| D | base_rendezvous_mgr.cc | 185 if (!IsLocalDevice(sess->worker_name(), parsed.src_device)) { in Send()
208 if (is_src && !IsLocalDevice(sess->worker_name(), parsed.src_device)) { in ValidateDevices()
249 Device* src_device; in SameWorkerRecvDone() local
250 Status s = sess->device_mgr()->LookupDevice(parsed.src_device, &src_device); in SameWorkerRecvDone()
288 src_device, dst_device, send_args.alloc_attrs, recv_args.alloc_attrs, &in, in SameWorkerRecvDone()
|
| D | collective_rma_distributed_test.cc | 110 << " src_device=" << request->src_device() in RecvBufAsync()
113 request->buf_rendezvous_key(), request->src_device(), in RecvBufAsync()
|
| D | worker.cc | 439 string local_name = DeviceNameUtils::LocalName(parsed.src_device); in PrepareRecvTensor()
|
| /external/llvm-project/lldb/test/API/commands/target/auto-install-main-executable/ |
| D | TestAutoInstallMainExecutable.py | 26 src_device = lldb.SBFileSpec(self.getBuildArtifact("a.device.out"))
28 err = lldb.remote_platform.Put(src_device, dest)
32 (src_device.GetFilename(), working_dir, err.GetCString()))
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | rendezvous_mgr.cc | 64 Device* src_device; in SameWorkerRecvDone() local
65 Status s = device_mgr->LookupDevice(parsed.src_device, &src_device); in SameWorkerRecvDone()
111 src_device, dst_device, send_args.alloc_attrs, recv_args.alloc_attrs, &in, in SameWorkerRecvDone()
|
| D | process_function_library_runtime.cc | 454 const string* src_device = AssignedOrRequestedDeviceName(*src_node); in PinArgsAndRets() local
458 << " src_device: " << *src_device in PinArgsAndRets()
460 while (src_device->empty() && colocation_group.empty() && in PinArgsAndRets()
467 src_device = AssignedOrRequestedDeviceName(*src_node); in PinArgsAndRets()
470 << " src_device: " << *src_device in PinArgsAndRets()
487 } else if (!src_device->empty() && can_use_src_node_device) { in PinArgsAndRets()
491 if (!DeviceNameUtils::ParseFullName(*src_device, &parsed)) { in PinArgsAndRets()
493 "Failed to parse explicit device specification ", *src_device); in PinArgsAndRets()
502 "Unable to find any devices for spec ", *src_device); in PinArgsAndRets()
548 *src_device, in PinArgsAndRets()
[all …]
|
| D | colocation_graph.cc | 230 [](const DeviceNameUtils::ParsedName& src_device, in IsEdgeFromCompositeDeviceToPhysicalDevice()
232 return src_device.has_type && dst_device.has_type && in IsEdgeFromCompositeDeviceToPhysicalDevice()
233 IsCompositeDevice(src_device.type) && in IsEdgeFromCompositeDeviceToPhysicalDevice()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr.cc | 243 if (!DeviceNameUtils::SplitDeviceName(parsed.src_device, &call->src_worker_, in RecvFromRemoteAsync()
245 s = errors::Internal(parsed.src_device, in RecvFromRemoteAsync()
|
| D | grpc_remote_worker.cc | 157 step_id, send_start_usec, end_usec, key, request->src_device(), in RecvBufAsync()
|
| D | grpc_worker_service.cc | 698 request->buf_rendezvous_key(), request->src_device(), in RecvBufAsync()
|
| /external/llvm-project/openmp/libomptarget/include/ |
| D | omptarget.h | 145 size_t src_offset, int dst_device, int src_device);
149 const size_t *src_dimensions, int dst_device, int src_device);
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.cc | 333 Device* src_device = h->device(); in SerializePackedHandle() local
336 (!ctx->OnSameTask(src_device, target_device)); in SerializePackedHandle()
344 op->add_handles()->mutable_remote_handle(), src_device, in SerializePackedHandle()
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.h | 156 std::shared_ptr<PjRtDevice> src_device) in TpuSharedBuffer()
158 device(std::move(src_device)), in TpuSharedBuffer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | fft_handler.cc | 286 [&](absl::Span<const int64> indices, int64 src_device) { in GetFinalFftUsingCollectivePermute() argument
290 src_dst_pairs.emplace_back(src_device, dst_device); in GetFinalFftUsingCollectivePermute()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 680 const string& src_device = src->assigned_device_name(); in AddControlFlow() local
683 if (src_device == dst_device) continue; in AddControlFlow()
762 const string& src_device = src->assigned_device_name(); in AddControlFlow() local
764 if (src_device != dst_device) { in AddControlFlow()
|
| /external/tensorflow/tensorflow/core/profiler/internal/gpu/ |
| D | cupti_tracer.cc | 366 size_t num_bytes, uint32 src_device, uint32 dst_device, bool async, in PopulateMemcpyCallbackEvent() argument
374 event.device_id = src_device; in PopulateMemcpyCallbackEvent()
444 uint32 dst_device = -1, src_device = -1; in AddP2PMemcpyEventUponApiExit() local
447 cupti_interface->GetDeviceId(p2p_params->srcContext, &src_device); in AddP2PMemcpyEventUponApiExit()
449 VLOG(3) << "Cuda P2P Memcpy API exit, src: " << src_device in AddP2PMemcpyEventUponApiExit()
452 PopulateMemcpyCallbackEvent(type, cbdata, num_bytes, src_device, in AddP2PMemcpyEventUponApiExit()
|
| /external/llvm-project/openmp/runtime/src/ |
| D | kmp_ftn_entry.h | 993 int dst_device, int src_device) { in FTN_TARGET_MEMCPY() argument
1001 int src_device) { in FTN_TARGET_MEMCPY_RECT() argument
|