| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr.cc | 63 AllocatorAttributes alloc_attrs, Device* dst_device, in Init()
130 Device* dst_device() const { return dst_device_; } in dst_device() function in tensorflow::__anona9090a8f0111::RpcRecvTensorCall
259 Device* dst_device; in RecvFromRemoteAsync() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker_cache_logger.cc | 89 const string& dst_device, in RecordRecvTensor()
99 const string& dst_device, in RecordDataTransfer()
|
| D | collective_rma_distributed_test.cc | 317 Device* dst_device = nullptr; in TEST_F() local
347 Device* dst_device = nullptr; in TEST_F() local
382 Device* dst_device = nullptr; in TEST_F() local
410 Device* dst_device = nullptr; in TEST_F() local
|
| D | base_rendezvous_mgr.cc | 255 Device* dst_device; in SameWorkerRecvDone() local
|
| /external/llvm-project/openmp/libomptarget/test/offloading/ |
| D | d2d_memcpy.c | 25 int dst_device = num_devices - 1; 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 | 63 StringPiece dst_device; member
|
| D | rendezvous.cc | 55 const string& dst_device, const string& name, in CreateKey()
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client_extension.cc | 146 [](PyTpuBuffer* buffer, std::shared_ptr<PjRtDevice> dst_device) { in PYBIND11_MODULE()
|
| D | tpu_client.cc | 391 std::shared_ptr<PjRtDevice> dst_device) { in CopyToDevice()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | rendezvous_mgr.cc | 70 Device* dst_device; in SameWorkerRecvDone() local
|
| D | replicate_per_replica_nodes.cc | 104 const string& dst_device = dst->assigned_device_name(); in ReplicateFromCompositeDeviceToRegularDevice() local
|
| D | colocation_graph.cc | 231 const DeviceNameUtils::ParsedName& dst_device) -> bool { in IsEdgeFromCompositeDeviceToPhysicalDevice()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | fft_handler.cc | 289 int64 dst_device = sharding.tile_assignment()(target_indices); in GetFinalFftUsingCollectivePermute() local
|
| D | spmd_partitioner.cc | 1341 int64 dst_device = target.tile_assignment()(indices); in ReshardWithCollectivePermute() local
|
| /external/llvm-project/openmp/runtime/src/ |
| D | kmp_ftn_entry.h | 993 int dst_device, int src_device) { in FTN_TARGET_MEMCPY()
1000 const size_t *dst_dimensions, const size_t *src_dimensions, int dst_device, in FTN_TARGET_MEMCPY_RECT()
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 507 string dst_device = 9; field
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 681 const string& dst_device = dst->assigned_device_name(); in AddControlFlow() local
763 const string& dst_device = dst->assigned_device_name(); in AddControlFlow() local
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 1294 PjRtDevice* dst_device, LocalDeviceState* dst_local_device, in CopyToDeviceHelper()
1355 PjRtDevice* dst_device) { in CopyToDevice()
|
| /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
|