| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Copy.cu | 186 Device src_device = iter.device(1); in copy_device_to_device() local
259 Device src_device = iter.device(1); in copy_requires_temporaries() local
281 static bool maybe_enable_p2p_access(Device dst_device, Device src_device) { in maybe_enable_p2p_access()
292 Device src_device = iter.device(1); in copy_kernel_cuda() local
|
| /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 | 262 Device* src_device; in SameWorkerRecvDone() local
|
| /external/pytorch/c10/core/ |
| D | CopyBytes.cpp | 35 Device src_device, in CopyBytes()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device_propagation.cc | 55 const string& src_device = AssignedOrRequestedDevice(*src); in UpdateDeviceFromInputs() local
|
| D | rendezvous_mgr.cc | 78 Device* src_device; in SameWorkerRecvDone() local
|
| D | colocation_graph.cc | 252 const DeviceNameUtils::ParsedName& dst_device) -> bool { in IsEdgeFromCompositeDeviceToPhysicalDevice()
|
| D | process_function_library_runtime.cc | 490 const string* src_device = AssignedOrRequestedDeviceName(*src_node); in PinArgsAndRets() local
|
| /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 | 151 std::string src_device = GetDevice(arg, func_op); in runOnOperation() local
|
| /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/compiler/xla/service/spmd/ |
| D | fft_handler.cc | 284 [&](absl::Span<const int64_t> indices, int64_t src_device) { in GetFinalFftUsingCollectivePermute()
|
| D | spmd_partitioner.cc | 1977 [&](absl::Span<const int64_t> indices, int64_t src_device) { in ReshardWithCollectivePermute()
|
| /external/ml_dtypes/ml_dtypes/tests/ |
| D | intn_test.cc | 409 Eigen::TensorMap<Eigen::Tensor<IntN, 1>, Eigen::Aligned> src_device( in TYPED_TEST() local
|
| D | float8_test.cc | 1099 Eigen::TensorMap<Eigen::Tensor<Float8, 1>, Eigen::Aligned> src_device( in TYPED_TEST() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.cc | 342 Device* src_device = h->device(); in SerializePackedHandle() local
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| D | worker.proto | 507 string src_device = 8; 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()
882 auto& src_device = *tensor_handle.GetAvailableDevice(); in CoreRTTensorHandleToFallbackTensorInternal() local
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition.cc | 679 const string& src_device = src->assigned_device_name(); in AddControlFlow() local
761 const string& src_device = src->assigned_device_name(); in AddControlFlow() local
|
| /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
|