Home
last modified time | relevance | path

Searched refs:Rendezvous (Results 1 – 25 of 46) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Drendezvous_test.cc40 const string key = Rendezvous::CreateKey( in TEST()
49 Rendezvous::ParsedKey parsed; in TEST()
50 TF_EXPECT_OK(Rendezvous::ParseKey(key, &parsed)); in TEST()
57 EXPECT_FALSE(Rendezvous::ParseKey("foo;bar;baz", &parsed).ok()); in TEST()
58 EXPECT_FALSE(Rendezvous::ParseKey("/job:mnist/replica:1/task:2/CPU:0;" in TEST()
63 Rendezvous::ParseKey(strings::StrCat(key, ";", key), &parsed).ok()); in TEST()
78 Rendezvous* rendez_;
98 Rendezvous::ParsedKey MakeKey(const string& name) { in MakeKey()
99 string s = Rendezvous::CreateKey("/job:mnist/replica:1/task:2/CPU:0", 7890, in MakeKey()
102 Rendezvous::ParsedKey k; in MakeKey()
[all …]
Drendezvous.cc36 Rendezvous::ParsedKey& Rendezvous::ParsedKey::operator=(const ParsedKey& b) { in operator =()
52 string Rendezvous::CreateKey(const string& src_device, uint64 src_incarnation, in CreateKey()
86 Status Rendezvous::ParseKey(StringPiece key, ParsedKey* out) { in ParseKey()
115 Rendezvous::~Rendezvous() {} in ~Rendezvous()
117 Status Rendezvous::Recv(const ParsedKey& key, const Args& recv_args, in Recv()
143 Status Rendezvous::Recv(const ParsedKey& key, const Args& args, Tensor* val, in Recv()
149 class LocalRendezvousImpl : public Rendezvous {
305 Rendezvous* NewLocalRendezvous() { return new LocalRendezvousImpl(); } in NewLocalRendezvous()
Drendezvous.h46 class Rendezvous : public core::RefCounted {
77 friend class Rendezvous;
124 ~Rendezvous() override;
130 Rendezvous* NewLocalRendezvous();
/external/tensorflow/tensorflow/core/common_runtime/
Drendezvous_util.cc20 Rendezvous* rendezvous, DeviceContext* device_context, in SendTensorsToRendezvous()
39 Rendezvous::ParsedKey parsed; in SendTensorsToRendezvous()
41 Rendezvous::Args rendez_args; in SendTensorsToRendezvous()
46 TF_RETURN_IF_ERROR(Rendezvous::ParseKey(keys[i], &parsed)); in SendTensorsToRendezvous()
54 Rendezvous* rendezvous, DeviceContext* device_context, in RecvOutputsFromRendezvousAsync()
70 std::tuple<string, Tensor*, Rendezvous::ParsedKey, AllocatorAttributes>> in RecvOutputsFromRendezvousAsync()
73 Rendezvous::ParsedKey parsed; in RecvOutputsFromRendezvousAsync()
74 Status s = Rendezvous::ParseKey(keys[i], &parsed); in RecvOutputsFromRendezvousAsync()
98 Rendezvous::ParsedKey parsed = std::get<2>(p); in RecvOutputsFromRendezvousAsync()
99 Rendezvous::Args rendez_args; in RecvOutputsFromRendezvousAsync()
[all …]
Dkernel_benchmark_testlib.cc45 Rendezvous* rendez) { in Benchmark()
122 return Rendezvous::CreateKey(send_device, send_device_incarnation, in GetRendezvousKey()
155 Rendezvous::ParsedKey parsed; in RunWithArgs()
156 TF_CHECK_OK(Rendezvous::ParseKey(p.first, &parsed)); in RunWithArgs()
157 TF_CHECK_OK(rendez_->Send(parsed, Rendezvous::Args(), p.second, false)); in RunWithArgs()
161 Rendezvous::ParsedKey parsed; in RunWithArgs()
162 TF_CHECK_OK(Rendezvous::ParseKey(key, &parsed)); in RunWithArgs()
163 TF_CHECK_OK(rendez_->Recv(parsed, Rendezvous::Args(), &unused, &is_dead)); in RunWithArgs()
172 Rendezvous::ParsedKey parsed; in RunWithArgs()
173 TF_CHECK_OK(Rendezvous::ParseKey(p.first, &parsed)); in RunWithArgs()
[all …]
Drendezvous_mgr.h40 class IntraProcessRendezvous : public Rendezvous {
46 Status Send(const ParsedKey& key, const Rendezvous::Args& args,
53 void RecvAsync(const ParsedKey& key, const Rendezvous::Args& args,
60 Rendezvous* local_; // Owns a Ref on this object.
73 Rendezvous::ParsedKey* parsed);
80 void SameWorkerRecvDone(const Rendezvous::ParsedKey& parsed,
81 const Rendezvous::Args& send_args,
82 const Rendezvous::Args& recv_args, const Tensor& in,
Drendezvous_mgr.cc40 const Rendezvous::Args& args, in Send()
53 Rendezvous::ParsedKey* parsed) { in ParseKey()
58 TF_RETURN_IF_ERROR(Rendezvous::ParseKey(key, parsed)); in ParseKey()
63 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& send_args, in SameWorkerRecvDone()
64 const Rendezvous::Args& recv_args, const Tensor& in, Tensor* out, in SameWorkerRecvDone()
117 const Rendezvous::Args& recv_args, in RecvAsync()
125 const Status& status, const Rendezvous::Args& send_args, in RecvAsync()
126 const Rendezvous::Args& recv_args, const Tensor& in, bool is_dead) { in RecvAsync()
Dgraph_runner.cc54 class SimpleRendezvous : public Rendezvous {
137 string full_key = Rendezvous::CreateKey("/device:CPU:0", 1, "/device:CPU:1", in Run()
139 Rendezvous::ParsedKey parsed; in Run()
140 TF_RETURN_IF_ERROR(Rendezvous::ParseKey(full_key, &parsed)); in Run()
141 TF_RETURN_IF_ERROR(rendez->Send(parsed, Rendezvous::Args(), in.second, in Run()
190 Rendezvous::CreateKey("/device:CPU:0", 1, "/device:CPU:1", in Run()
192 Rendezvous::ParsedKey parsed; in Run()
193 TF_RETURN_IF_ERROR(Rendezvous::ParseKey(output_key, &parsed)); in Run()
197 rendez->Recv(parsed, Rendezvous::Args(), &output_tensor, &is_dead)); in Run()
Drendezvous_util.h33 Rendezvous* rendezvous, DeviceContext* device_context,
42 Rendezvous* rendezvous, DeviceContext* device_context,
47 Status RecvOutputsFromRendezvous(Rendezvous* rendezvous, NamedTensors* out,
48 const Rendezvous::Args& args);
Dexecutor.h85 Rendezvous* rendezvous = nullptr;
165 ExecutorBarrier(size_t num, Rendezvous* r, StatusCallback done) in ExecutorBarrier()
177 Rendezvous* rendez_ = nullptr;
186 Rendezvous* error_rendez = nullptr; in WhenDone()
Dkernel_benchmark_testlib.h42 Rendezvous* rendez = nullptr);
59 Rendezvous* rendez_ = nullptr;
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Drpc_rendezvous_mgr_test.cc43 Rendezvous::ParsedKey MakeKey(const string& s) { in MakeKey()
44 Rendezvous::ParsedKey key; in MakeKey()
45 CHECK(Rendezvous::ParseKey(s, &key).ok()); in MakeKey()
86 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
93 Rendezvous::Args args; in TEST_F()
106 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
119 Rendezvous::Args args; in TEST_F()
133 Rendezvous::Args args; in TEST_F()
140 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
148 Rendezvous::Args args; in TEST_F()
[all …]
Drpc_rendezvous_mgr.cc46 void RecvFromRemoteAsync(const Rendezvous::ParsedKey& parsed,
47 const Rendezvous::Args& args,
63 const Rendezvous::Args& recv_args, Rendezvous::DoneCallback done) { in Init()
121 const Rendezvous::Args& recv_args() const { return recv_args_; } in recv_args()
122 const Rendezvous::DoneCallback& done() const { return done_; } in done()
153 Rendezvous::Args recv_args_;
154 Rendezvous::DoneCallback done_;
208 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& recv_args, in RecvFromRemoteAsync()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dbase_rendezvous_mgr.h78 void RecvLocalAsync(int64 step_id, const Rendezvous::ParsedKey& parsed,
79 Rendezvous::DoneCallback done) override;
82 Status RecvLocal(int64 step_id, const Rendezvous::ParsedKey& parsed,
128 Status Send(const ParsedKey& key, const Rendezvous::Args& args,
135 void RecvAsync(const ParsedKey& key, const Rendezvous::Args& args,
153 virtual void RecvFromRemoteAsync(const Rendezvous::ParsedKey& parsed,
154 const Rendezvous::Args& args,
178 Rendezvous* local_; // Owns a Ref on this object.
205 Status ValidateDevices(const Rendezvous::ParsedKey& parsed, bool is_src);
211 void SameWorkerRecvDone(const Rendezvous::ParsedKey& parsed,
[all …]
Dbase_rendezvous_mgr.cc38 static void StartAbortRendevous(Rendezvous* rendez, const Status& s) { in StartAbortRendevous()
69 const Rendezvous::ParsedKey& parsed, in RecvLocalAsync()
70 Rendezvous::DoneCallback done) { in RecvLocalAsync()
73 Rendezvous::DoneCallback done_cb = std::bind( in RecvLocalAsync()
74 [rendez](Rendezvous::DoneCallback done, in RecvLocalAsync()
76 const Status& s, const Rendezvous::Args& send_args, in RecvLocalAsync()
77 const Rendezvous::Args& recv_args, const Tensor& v, bool dead) { in RecvLocalAsync()
86 const Rendezvous::ParsedKey& parsed, in RecvLocal()
92 const Rendezvous::Args& send_args, in RecvLocal()
93 const Rendezvous::Args& recv_args, in RecvLocal()
[all …]
Dexecutor_test.cc77 Status Run(Rendezvous* rendez) { in Run()
91 Rendezvous* rendez_ = nullptr;
131 Rendezvous::ParsedKey Key(const string& sender, const uint64 incarnation, in Key()
133 Rendezvous::ParsedKey result; in Key()
135 Rendezvous::ParseKey(Rendezvous::CreateKey(sender, incarnation, receiver, in Key()
153 Rendezvous::Args args; in TEST_F()
184 Rendezvous::Args args; in TEST_F()
235 Rendezvous::Args args; in TEST_F()
269 Rendezvous* rendez = NewLocalRendezvous(); in TEST_F()
271 Rendezvous::Args args; in TEST_F()
[all …]
Drendezvous_mgr_interface.h39 class RemoteRendezvous : public Rendezvous {
81 const Rendezvous::ParsedKey& parsed,
82 Rendezvous::DoneCallback done) = 0;
85 virtual Status RecvLocal(int64 step_id, const Rendezvous::ParsedKey& parsed,
/external/tensorflow/tensorflow/core/kernels/
Dsendrecv_ops.cc71 OP_REQUIRES_OK(ctx, Rendezvous::ParseKey(parsed_key_.buf_, &parsed_key_)); in SendOp()
86 Rendezvous::Args args; in Compute()
98 Rendezvous::ParsedKey in_loop_parsed; in Compute()
102 Rendezvous::ParseKey(in_loop_parsed.buf_, &in_loop_parsed)); in Compute()
139 OP_REQUIRES_OK(ctx, Rendezvous::ParseKey(parsed_key_.buf_, &parsed_key_)); in RecvOp()
146 Rendezvous::DoneCallback make_recv_callback(OpKernelContext* ctx, in make_recv_callback()
152 const Status& s, const Rendezvous::Args& send_args, in make_recv_callback()
153 const Rendezvous::Args& recv_args, const Tensor& val, in make_recv_callback()
176 Rendezvous::Args args; in ComputeAsync()
186 Rendezvous::ParsedKey in_loop_parsed; in ComputeAsync()
[all …]
Dsendrecv_ops.h31 Rendezvous::ParsedKey parsed_key_;
44 Rendezvous::ParsedKey parsed_key_;
/external/tensorflow/tensorflow/contrib/mpi/
Dmpi_rendezvous_mgr.cc63 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& recv_args, in RecvFromRemoteAsync()
157 Rendezvous::ParsedKey parsed; in AddRequest()
158 TF_CHECK_OK(Rendezvous::ParseKey(key, &parsed)); in AddRequest()
162 const Rendezvous::Args& send_args, in AddRequest()
163 const Rendezvous::Args& recv_args, in AddRequest()
198 Rendezvous::DoneCallback done_cb = in AddRequest()
200 const Status& status, const Rendezvous::Args& send_args, in AddRequest()
201 const Rendezvous::Args& recv_args, const Tensor& val, bool is_dead) { in AddRequest()
Dmpi_rendezvous_mgr.h72 void Init(const Rendezvous::ParsedKey& parsed, const int64 step_id, in Init()
92 Rendezvous::DoneCallback done_;
106 void Init(const Rendezvous::ParsedKey& parsed, const int64 step_id) { in Init()
125 void RecvFromRemoteAsync(const Rendezvous::ParsedKey& parsed,
126 const Rendezvous::Args& args,
163 const Status&, const Rendezvous::Args&, const Rendezvous::Args&,
/external/tensorflow/tensorflow/contrib/verbs/
Drdma_rendezvous_mgr.cc36 void RecvFromRemoteAsync(const Rendezvous::ParsedKey& parsed,
37 const Rendezvous::Args& args,
48 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& recv_args, in RecvFromRemoteAsync()
Drdma.h236 typedef Rendezvous::DoneCallback RecvDoneCallback;
241 const Rendezvous::Args recv_args,
294 Rendezvous::Args recv_args_;
338 void RecvHandler(Rendezvous::ParsedKey parsed,
339 const Rendezvous::Args& send_args,
340 const Rendezvous::Args& recv_args, const Tensor& in,
346 Status PrepareRecvTensor(const Rendezvous::ParsedKey& parsed,
433 const Rendezvous::Args recv_args,
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_rendezvous_mgr.cc43 const Rendezvous::Args& recv_args, int64 step_id, in GdrRecvTensorCall()
109 const Rendezvous::Args& recv_args() const { return recv_args_; } in recv_args()
118 Rendezvous::Args recv_args_;
134 void RecvFromRemoteAsync(const Rendezvous::ParsedKey& parsed, in RecvFromRemoteAsync()
135 const Rendezvous::Args& recv_args, in RecvFromRemoteAsync()
Dgdr_worker.cc62 Rendezvous::ParsedKey parsed; in GrpcRecvTensorAsync()
63 s = Rendezvous::ParseKey(key, &parsed); in GrpcRecvTensorAsync()
82 const Status& status, const Rendezvous::Args& send_args, in GrpcRecvTensorAsync()
83 const Rendezvous::Args&, const Tensor& val, const bool is_dead) { in GrpcRecvTensorAsync()

12