/external/tensorflow/tensorflow/core/distributed_runtime/ |
D | worker_cache_logger.cc | 88 const string& src_device, in RecordRecvTensor() 98 const string& src_device, in RecordDataTransfer()
|
D | base_rendezvous_mgr.cc | 249 Device* src_device; in SameWorkerRecvDone() local
|
/external/llvm-project/openmp/libomptarget/test/offloading/ |
D | d2d_memcpy.c | 24 const int src_device = 0; in main() local
|
/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() 192 const size_t *src_dimensions, int dst_device, int src_device) { in omp_target_memcpy_rect()
|
/external/tensorflow/tensorflow/core/framework/ |
D | rendezvous.h | 60 StringPiece src_device; member
|
D | rendezvous.cc | 54 string Rendezvous::CreateKey(const string& src_device, uint64 src_incarnation, in CreateKey()
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | rendezvous_mgr.cc | 64 Device* src_device; in SameWorkerRecvDone() local
|
D | colocation_graph.cc | 231 const DeviceNameUtils::ParsedName& dst_device) -> bool { in IsEdgeFromCompositeDeviceToPhysicalDevice()
|
D | process_function_library_runtime.cc | 454 const string* src_device = AssignedOrRequestedDeviceName(*src_node); in PinArgsAndRets() local
|
/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
D | fft_handler.cc | 286 [&](absl::Span<const int64> indices, int64 src_device) { in GetFinalFftUsingCollectivePermute()
|
D | spmd_partitioner.cc | 1340 [&](absl::Span<const int64> indices, int64 src_device) { in ReshardWithCollectivePermute()
|
/external/llvm-project/openmp/runtime/src/ |
D | kmp_ftn_entry.h | 993 int dst_device, int src_device) { in FTN_TARGET_MEMCPY() 1001 int src_device) { in FTN_TARGET_MEMCPY_RECT()
|
/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
D | remote_copy_node.cc | 333 Device* src_device = h->device(); in SerializePackedHandle() local
|
/external/tensorflow/tensorflow/core/protobuf/ |
D | worker.proto | 505 string src_device = 8; field
|
/external/tensorflow/tensorflow/core/graph/ |
D | graph_partition.cc | 680 const string& src_device = src->assigned_device_name(); in AddControlFlow() local 762 const string& src_device = src->assigned_device_name(); in AddControlFlow() local
|
/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() 444 uint32 dst_device = -1, src_device = -1; in AddP2PMemcpyEventUponApiExit() local
|
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
D | distributed_tpu_rewrite_pass.cc | 1156 const string& src_device = AssignedOrRequestedDevice(src); in PlaceOpsOnTPU() local
|