Home
last modified time | relevance | path

Searched defs:dst_device (Results 1 – 25 of 27) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DCopy.cu185 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
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dcollective_rma_distributed_test.cc343 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
H A Dworker_cache_logger.cc89 const string& dst_device, in RecordRecvTensor()
99 const string& dst_device, in RecordDataTransfer()
H A Dbase_rendezvous_mgr.cc268 Device* dst_device; in SameWorkerRecvDone() local
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
H A Drpc_rendezvous_mgr.cc61 AllocatorAttributes alloc_attrs, Device* dst_device, in Init()
128 Device* dst_device() const { return dst_device_; } in dst_device() function in tensorflow::__anon0b28f0be0111::RpcRecvTensorCall
257 Device* dst_device; in RecvFromRemoteAsync() local
/aosp_15_r20/external/pytorch/c10/core/
H A DCopyBytes.cpp37 Device dst_device, in CopyBytes()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
H A Dtensor_util.cc31 const tfrt::Device& src_device, const tfrt::Device& dst_device) { in TransferTensorToDevice()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
H A Dcross_device_transfer.cc136 std::string dst_device = GetDevice(op); in runOnOperation() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Drendezvous.h63 StringPiece dst_device; member
H A Drendezvous.cc55 const string& dst_device, const string& name, in CreateKey()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dreplicate_per_replica_nodes.cc117 const string& dst_device = dst->assigned_device_name(); in ReplicateFromCompositeDeviceToRegularDevice() local
H A Drendezvous_mgr.cc84 Device* dst_device; in SameWorkerRecvDone() local
H A Dcolocation_graph.cc252 const DeviceNameUtils::ParsedName& dst_device) -> bool { in IsEdgeFromCompositeDeviceToPhysicalDevice()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
H A Dtpu_client_extension.cc171 [](PyTpuBuffer* buffer, std::shared_ptr<PjRtDevice> dst_device) { in PYBIND11_MODULE()
H A Dtpu_client.cc402 std::shared_ptr<PjRtDevice> dst_device) { in CopyToDevice()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_kernels.cc842 const tfrt::Device& dst_device) -> tfrt::AsyncValueRef<tfrt::Tensor> { in CoreRTTensorHandleToFallbackTensorInternal()
863 auto dst_device = in CoreRTTensorHandleToFallbackTensorInternal() local
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A Dgraph_partition.cc680 const string& dst_device = dst->assigned_device_name(); in AddControlFlow() local
762 const string& dst_device = dst->assigned_device_name(); in AddControlFlow() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dfft_handler.cc287 int64_t dst_device = sharding.tile_assignment()(target_indices); in GetFinalFftUsingCollectivePermute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dworker.proto509 string dst_device = 9; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_stream_executor_client.cc1418 PjRtDevice* dst_device, LocalDeviceState* dst_local_device, in CopyToDeviceHelper()
1478 PjRtDevice* dst_device) { in CopyToDevice()
H A Dpjrt_c_api_client.cc728 PjRtDevice* dst_device) { in CopyToDevice()
H A Dtfrt_cpu_pjrt_client.cc1016 PjRtDevice* dst_device) { in CopyToDevice()
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/backends/gpu/
H A Dcupti_tracer.cc390 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/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_buffer.cc605 [](PyBuffer::object self, const ClientAndPtr<PjRtDevice>& dst_device) { in RegisterTypes()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/c/
H A Dpjrt_c_api.h698 PJRT_Device* dst_device; member

12