| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Copy.cu | 185 Device dst_device = iter.device(0); in copy_device_to_device() local
258 Device dst_device = iter.device(0); in copy_requires_temporaries() local
281 static bool maybe_enable_p2p_access(Device dst_device, Device src_device) { in maybe_enable_p2p_access()
291 Device dst_device = iter.device(0); in copy_kernel_cuda() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr.cc | 61 AllocatorAttributes alloc_attrs, Device* dst_device, in Init()
128 Device* dst_device() const { return dst_device_; } in dst_device() function in tensorflow::__anonb8b05b3e0111::RpcRecvTensorCall
257 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 | 343 void MaybeSetGPUDevice(Device* dst_device) { in MaybeSetGPUDevice()
388 Device* dst_device = nullptr; in TEST_P() local
419 Device* dst_device = nullptr; in TEST_P() local
455 Device* dst_device = nullptr; in TEST_P() local
492 Device* dst_device = nullptr; in TEST_P() local
524 Device* dst_device = nullptr; in TEST_P() local
|
| D | base_rendezvous_mgr.cc | 268 Device* dst_device; in SameWorkerRecvDone() local
|
| /external/pytorch/c10/core/ |
| D | CopyBytes.cpp | 37 Device dst_device, in CopyBytes()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | tensor_util.cc | 31 const tfrt::Device& src_device, const tfrt::Device& dst_device) { in TransferTensorToDevice()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | cross_device_transfer.cc | 136 std::string dst_device = GetDevice(op); in runOnOperation() local
|
| /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/core/common_runtime/ |
| D | replicate_per_replica_nodes.cc | 117 const string& dst_device = dst->assigned_device_name(); in ReplicateFromCompositeDeviceToRegularDevice() local
|
| D | rendezvous_mgr.cc | 84 Device* dst_device; in SameWorkerRecvDone() local
|
| D | colocation_graph.cc | 252 const DeviceNameUtils::ParsedName& dst_device) -> bool { in IsEdgeFromCompositeDeviceToPhysicalDevice()
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client_extension.cc | 171 [](PyTpuBuffer* buffer, std::shared_ptr<PjRtDevice> dst_device) { in PYBIND11_MODULE()
|
| D | tpu_client.cc | 402 std::shared_ptr<PjRtDevice> dst_device) { in CopyToDevice()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | fft_handler.cc | 287 int64_t dst_device = sharding.tile_assignment()(target_indices); in GetFinalFftUsingCollectivePermute() local
|
| /external/ml_dtypes/ml_dtypes/tests/ |
| D | intn_test.cc | 411 Eigen::TensorMap<Eigen::Tensor<DestType, 1>, Eigen::Aligned> dst_device( in TYPED_TEST() local
|
| D | float8_test.cc | 1101 Eigen::TensorMap<Eigen::Tensor<DestType, 1>, Eigen::Aligned> dst_device( in TYPED_TEST() local
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 509 string dst_device = 9; field
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_kernels.cc | 842 const tfrt::Device& dst_device) -> tfrt::AsyncValueRef<tfrt::Tensor> { in CoreRTTensorHandleToFallbackTensorInternal()
863 auto dst_device = in CoreRTTensorHandleToFallbackTensorInternal() local
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 680 const string& dst_device = dst->assigned_device_name(); in AddControlFlow() local
762 const string& dst_device = dst->assigned_device_name(); in AddControlFlow() local
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_c_api_client.cc | 728 PjRtDevice* dst_device) { in CopyToDevice()
|
| D | pjrt_stream_executor_client.cc | 1418 PjRtDevice* dst_device, LocalDeviceState* dst_local_device, in CopyToDeviceHelper()
1478 PjRtDevice* dst_device) { in CopyToDevice()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | py_buffer.cc | 605 [](PyBuffer::object self, const ClientAndPtr<PjRtDevice>& dst_device) { in RegisterTypes()
|
| /external/tensorflow/tensorflow/core/profiler/backends/gpu/ |
| D | cupti_tracer.cc | 390 size_t num_bytes, uint32 src_device, uint32 dst_device, bool async, in PopulateMemcpyCallbackEvent()
471 uint32 dst_device = -1, src_device = -1; in AddP2PMemcpyEventUponApiExit() local
|