Home
last modified time | relevance | path

Searched defs:recv_device (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_copy_node.cc78 Device* recv_device, uint64 recv_op_id) in RemoteCopyNode()
244 Device* recv_device = recv_device_; in RunRemoteRecv() local
274 Device* recv_device = ctx_->CanonicalDevice(recv_device_); in StartRecv() local
394 Device* recv_device = recv_device_; in StartSendPackedHandle() local
449 Device* recv_device = recv_device_; in StartRemoteSendTensor() local
/external/tensorflow/tensorflow/core/kernels/
Dsendrecv_ops.cc30 const string& recv_device, in GetRendezvousKeyPrefix()
61 string recv_device; in SendOp() local
140 string recv_device; in RecvOp() local
/external/tensorflow/tensorflow/core/common_runtime/
Dkernel_benchmark_testlib.cc150 string recv_device; in GetRendezvousKey() local
Dimmutable_executor_state.cc114 string recv_device; in Initialize() local
Dstep_stats_collector.cc92 string recv_device; in Done() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_group_metadata.cc486 auto recv_device = GetInstructionDevice(*channel.recv); in VerifyChannelInstructions() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler_test.cc3031 const string recv_device = kCPU1; in TEST_F() local
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_session.cc404 string recv_device; in TrackFeedsAndFetches() local