Home
last modified time | relevance | path

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

123

/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 {
307 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.cc23 Rendezvous* rendezvous, DeviceContext* device_context, in SendTensorsToRendezvous()
42 Rendezvous::ParsedKey parsed; in SendTensorsToRendezvous()
44 Rendezvous::Args rendez_args; in SendTensorsToRendezvous()
49 TF_RETURN_IF_ERROR(Rendezvous::ParseKey(keys[i], &parsed)); in SendTensorsToRendezvous()
57 Rendezvous* rendezvous, DeviceContext* device_context, in RecvOutputsFromRendezvousAsync()
73 std::tuple<string, Tensor*, Rendezvous::ParsedKey, AllocatorAttributes>> in RecvOutputsFromRendezvousAsync()
76 Rendezvous::ParsedKey parsed; in RecvOutputsFromRendezvousAsync()
77 Status s = Rendezvous::ParseKey(keys[i], &parsed); in RecvOutputsFromRendezvousAsync()
95 Rendezvous::ParsedKey parsed = std::get<2>(p); in RecvOutputsFromRendezvousAsync()
96 Rendezvous::Args rendez_args; in RecvOutputsFromRendezvousAsync()
[all …]
Dkernel_benchmark_testlib.cc47 Rendezvous* rendez, const char* executor_type) { in Benchmark()
127 return Rendezvous::CreateKey(send_device, send_device_incarnation, in GetRendezvousKey()
160 Rendezvous::ParsedKey parsed; in RunWithArgs()
161 TF_CHECK_OK(Rendezvous::ParseKey(p.first, &parsed)); in RunWithArgs()
162 TF_CHECK_OK(rendez_->Send(parsed, Rendezvous::Args(), p.second, false)); in RunWithArgs()
166 Rendezvous::ParsedKey parsed; in RunWithArgs()
167 TF_CHECK_OK(Rendezvous::ParseKey(key, &parsed)); in RunWithArgs()
168 TF_CHECK_OK(rendez_->Recv(parsed, Rendezvous::Args(), &unused, &is_dead)); in RunWithArgs()
177 Rendezvous::ParsedKey parsed; in RunWithArgs()
178 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()
128 const Rendezvous::Args& send_args, in RecvAsync()
129 const Rendezvous::Args& recv_args, const Tensor& in, in RecvAsync()
Dgraph_runner.cc48 class SimpleRendezvous : public Rendezvous {
135 string full_key = Rendezvous::CreateKey("/device:CPU:0", 1, "/device:CPU:1", in Run()
137 Rendezvous::ParsedKey parsed; in Run()
138 TF_RETURN_IF_ERROR(Rendezvous::ParseKey(full_key, &parsed)); in Run()
139 TF_RETURN_IF_ERROR(rendez->Send(parsed, Rendezvous::Args(), in.second, in Run()
193 Rendezvous::CreateKey("/device:CPU:0", 1, "/device:CPU:1", in Run()
195 Rendezvous::ParsedKey parsed; in Run()
196 TF_RETURN_IF_ERROR(Rendezvous::ParseKey(output_key, &parsed)); in Run()
200 rendez->Recv(parsed, Rendezvous::Args(), &output_tensor, &is_dead)); in Run()
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_util.h34 Rendezvous* rendezvous, DeviceContext* device_context,
43 Rendezvous* rendezvous, DeviceContext* device_context,
48 Status RecvOutputsFromRendezvous(Rendezvous* rendezvous, NamedTensors* out,
49 const Rendezvous::Args& args);
Dexecutor.h86 Rendezvous* rendezvous = nullptr;
160 ExecutorBarrier(size_t num, Rendezvous* r, StatusCallback done) in ExecutorBarrier()
172 Rendezvous* rendez_ = nullptr;
180 Rendezvous* error_rendez = nullptr; in WhenDone()
/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()
88 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
95 Rendezvous::Args args; in TEST_F()
108 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
121 Rendezvous::Args args; in TEST_F()
135 Rendezvous::Args args; in TEST_F()
142 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
150 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()
130 const Rendezvous::Args& recv_args() const { return recv_args_; } in recv_args()
131 const Rendezvous::DoneCallback& done() const { return done_; } in done()
162 Rendezvous::Args recv_args_;
163 Rendezvous::DoneCallback done_;
217 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 …]
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 …]
/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()
104 const Rendezvous::Args& recv_args() const { return recv_args_; } in recv_args()
113 Rendezvous::Args recv_args_;
129 void RecvFromRemoteAsync(const Rendezvous::ParsedKey& parsed, in RecvFromRemoteAsync()
130 const Rendezvous::Args& recv_args, in RecvFromRemoteAsync()
Dgdr_worker.cc61 Rendezvous::ParsedKey parsed; in GrpcRecvTensorAsync()
62 s = Rendezvous::ParseKey(key, &parsed); in GrpcRecvTensorAsync()
81 const Status& status, const Rendezvous::Args& send_args, in GrpcRecvTensorAsync()
82 const Rendezvous::Args&, const Tensor& val, const bool is_dead) { in GrpcRecvTensorAsync()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.h82 Rendezvous* rendezvous);
87 Rendezvous* rendezvous);
160 Rendezvous* GetRendezvous() const { return rendezvous_; } in GetRendezvous()
222 const gtl::FlatMap<string, uint64>& remote_contexts, Rendezvous* r,
268 Rendezvous* rendezvous_;
/external/tensorflow/tensorflow/core/kernels/data/
Dsingle_threaded_executor_test.cc75 Status Run(Rendezvous* rendez) { in Run()
92 Rendezvous* rendez_ = nullptr;
130 Rendezvous::ParsedKey Key(const string& sender, const uint64 incarnation, in Key()
132 Rendezvous::ParsedKey result; in Key()
134 Rendezvous::ParseKey(Rendezvous::CreateKey(sender, incarnation, receiver, in Key()

123