| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | worker_cache_logger.cc | 88 const string& src_device, in RecordRecvTensor()
98 const string& src_device, in RecordDataTransfer()
|
| H A D | base_rendezvous_mgr.cc | 262 Device* src_device; in SameWorkerRecvDone() local
|
| /aosp_15_r20/tools/test/connectivity/acts_tests/acts_contrib/test_utils/wifi/wifi_performance_test_utils/ |
| D | __init__.py | 249 def get_ping_stats(src_device, dest_address, ping_duration, ping_interval, argument
303 def get_ping_stats_nb(src_device, dest_address, ping_duration, ping_interval, argument
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | CopyBytes.cpp | 35 Device src_device, in CopyBytes()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | device_propagation.cc | 55 const string& src_device = AssignedOrRequestedDevice(*src); in UpdateDeviceFromInputs() local
|
| H A D | rendezvous_mgr.cc | 78 Device* src_device; in SameWorkerRecvDone() local
|
| H A D | colocation_graph.cc | 252 const DeviceNameUtils::ParsedName& dst_device) -> bool { in IsEdgeFromCompositeDeviceToPhysicalDevice()
|
| H A D | process_function_library_runtime.cc | 490 const string* src_device = AssignedOrRequestedDeviceName(*src_node); in PinArgsAndRets() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| H A D | tensor_util.cc | 31 const tfrt::Device& src_device, const tfrt::Device& dst_device) { in TransferTensorToDevice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| H A D | cross_device_transfer.cc | 151 std::string src_device = GetDevice(arg, func_op); in runOnOperation() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | rendezvous.h | 60 StringPiece src_device; member
|
| H A D | rendezvous.cc | 54 string Rendezvous::CreateKey(const string& src_device, uint64 src_incarnation, in CreateKey()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | fft_handler.cc | 284 [&](absl::Span<const int64_t> indices, int64_t src_device) { in GetFinalFftUsingCollectivePermute()
|
| H A D | spmd_partitioner.cc | 1977 [&](absl::Span<const int64_t> indices, int64_t src_device) { in ReshardWithCollectivePermute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| H A D | remote_copy_node.cc | 342 Device* src_device = h->device(); in SerializePackedHandle() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/ |
| H A D | worker.proto | 507 string src_device = 8; field
|
| /aosp_15_r20/tools/test/connectivity/acts_tests/tests/google/wifi/ |
| D | WifiManagerTest.py | 553 def verify_traffic_between_devices(self,dest_device,src_device,num_of_tries=2): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/backends/gpu/ |
| H A 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
|
| /aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/ |
| D | worker.pb.h | 9613 inline const std::string& RecvBufRequest::src_device() const { in src_device() function
9644 inline void RecvBufRequest::set_allocated_src_device(std::string* src_device) { in set_allocated_src_device()
|