Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/core/framework/
Drendezvous_test.cc42 const string key = Rendezvous::CreateKey( in TEST()
51 Rendezvous::ParsedKey parsed; in TEST()
52 TF_EXPECT_OK(Rendezvous::ParseKey(key, &parsed)); in TEST()
59 EXPECT_FALSE(Rendezvous::ParseKey("foo;bar;baz", &parsed).ok()); in TEST()
60 EXPECT_FALSE(Rendezvous::ParseKey("/job:mnist/replica:1/task:2/CPU:0;" in TEST()
65 Rendezvous::ParseKey(strings::StrCat(key, ";", key), &parsed).ok()); in TEST()
80 Rendezvous* rendez_;
100 Rendezvous::ParsedKey MakeKey(const string& name) { in MakeKey()
101 string s = Rendezvous::CreateKey("/job:mnist/replica:1/task:2/CPU:0", 7890, in MakeKey()
104 Rendezvous::ParsedKey k; in MakeKey()
[all …]
Dlocal_rendezvous.cc35 Item(Rendezvous::Args send_args, const Tensor& value, bool is_dead) in Item()
41 Item(Rendezvous::Args recv_args, Rendezvous::DoneCallback waiter, in Item()
59 const Rendezvous::Args args;
73 ManualConstructor<Rendezvous::DoneCallback> waiter;
79 Item(Rendezvous::Args args, Type type) : args(args), type(type) { in Item()
108 Status LocalRendezvous::Send(const Rendezvous::ParsedKey& key, in Send()
109 const Rendezvous::Args& send_args, in Send()
156 void LocalRendezvous::RecvAsync(const Rendezvous::ParsedKey& key, in RecvAsync()
157 const Rendezvous::Args& recv_args, in RecvAsync()
158 Rendezvous::DoneCallback done) { in RecvAsync()
[all …]
Dlocal_rendezvous.h41 Status Send(const Rendezvous::ParsedKey& key,
42 const Rendezvous::Args& send_args, const Tensor& val,
44 void RecvAsync(const Rendezvous::ParsedKey& key,
45 const Rendezvous::Args& recv_args,
46 Rendezvous::DoneCallback done);
Drendezvous.cc38 Rendezvous::ParsedKey& Rendezvous::ParsedKey::operator=(const ParsedKey& b) { in operator =()
54 string Rendezvous::CreateKey(const string& src_device, uint64 src_incarnation, in CreateKey()
88 Status Rendezvous::ParseKey(StringPiece key, ParsedKey* out) { in ParseKey()
152 class LocalRendezvousWrapper : public Rendezvous {
175 Rendezvous* NewLocalRendezvous() { return new LocalRendezvousWrapper; } in NewLocalRendezvous()
Drendezvous.h72 friend class Rendezvous;
128 class Rendezvous : public RendezvousInterface, public core::RefCounted {
143 Rendezvous* NewLocalRendezvous();
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Drpc_rendezvous_mgr_test.cc44 Rendezvous::ParsedKey MakeKey(const string& s) { in MakeKey()
45 Rendezvous::ParsedKey key; in MakeKey()
46 CHECK(Rendezvous::ParseKey(s, &key).ok()); in MakeKey()
93 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
100 Rendezvous::Args args; in TEST_F()
113 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
126 Rendezvous::Args args; in TEST_F()
140 Rendezvous::Args args; in TEST_F()
147 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
162 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()
157 Rendezvous::Args recv_args_;
158 Rendezvous::DoneCallback done_;
212 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,
163 void RegisterCall(BaseRecvTensorCall* call, const Rendezvous::Args& args);
178 Rendezvous* local_; // Owns a Ref on this object.
212 Status ValidateDevices(const Rendezvous::ParsedKey& parsed, bool is_src);
[all …]
Dbase_rendezvous_mgr.cc40 static void StartAbortRendevous(Rendezvous* rendez, const Status& s) { in StartAbortRendevous()
71 const Rendezvous::ParsedKey& parsed, in RecvLocalAsync()
72 Rendezvous::DoneCallback done) { in RecvLocalAsync()
75 const Status& s, const Rendezvous::Args& send_args, in RecvLocalAsync()
76 const Rendezvous::Args& recv_args, const Tensor& v, in RecvLocalAsync()
85 const Rendezvous::ParsedKey& parsed, in RecvLocal()
91 const Rendezvous::Args& send_args, in RecvLocal()
92 const Rendezvous::Args& recv_args, in RecvLocal()
104 Rendezvous* rendez = nullptr; in Cleanup()
119 std::vector<Rendezvous*> rendezs; in CleanupAll()
[all …]
Drendezvous_mgr_interface.h39 class RemoteRendezvous : public Rendezvous {
84 const Rendezvous::ParsedKey& parsed,
85 Rendezvous::DoneCallback done) = 0;
88 virtual Status RecvLocal(int64 step_id, const Rendezvous::ParsedKey& parsed,
/external/tensorflow/tensorflow/core/common_runtime/
Dkernel_benchmark_testlib.cc49 Rendezvous* rendez, const char* executor_type) { in Benchmark()
132 return Rendezvous::CreateKey(send_device, send_device_incarnation, in GetRendezvousKey()
154 Rendezvous::ParsedKey parsed; in RunWithRendezvousArgs()
155 TF_CHECK_OK(Rendezvous::ParseKey(p.first, &parsed)); in RunWithRendezvousArgs()
156 TF_CHECK_OK(rendez_->Send(parsed, Rendezvous::Args(), p.second, false)); in RunWithRendezvousArgs()
160 Rendezvous::ParsedKey parsed; in RunWithRendezvousArgs()
161 TF_CHECK_OK(Rendezvous::ParseKey(key, &parsed)); in RunWithRendezvousArgs()
162 TF_CHECK_OK(rendez_->Recv(parsed, Rendezvous::Args(), &unused, &is_dead)); in RunWithRendezvousArgs()
171 Rendezvous::ParsedKey parsed; in RunWithRendezvousArgs()
172 TF_CHECK_OK(Rendezvous::ParseKey(p.first, &parsed)); in RunWithRendezvousArgs()
[all …]
Drendezvous_util.cc42 Rendezvous::ParsedKey parsed; in SendTensorsToRendezvous()
44 Rendezvous::Args rendez_args; in SendTensorsToRendezvous()
49 TF_RETURN_IF_ERROR(Rendezvous::ParseKey(keys[i], &parsed)); in SendTensorsToRendezvous()
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()
103 const Rendezvous::Args& send_args, in RecvOutputsFromRendezvousAsync()
104 const Rendezvous::Args& recv_args, in RecvOutputsFromRendezvousAsync()
[all …]
Drendezvous_mgr.cc37 const Rendezvous::ParsedKey& parsed, in SameWorkerRecvDone()
38 const Rendezvous::Args& send_args, in SameWorkerRecvDone()
39 const Rendezvous::Args& recv_args, const Tensor& in, in SameWorkerRecvDone()
111 const Rendezvous::Args& recv_args, in IntraProcessRecvAsyncImpl()
120 const Status& status, const Rendezvous::Args& send_args, in IntraProcessRecvAsyncImpl()
121 const Rendezvous::Args& recv_args, const Tensor& in, in IntraProcessRecvAsyncImpl()
152 const Rendezvous::Args& args, in Send()
160 const Rendezvous::Args& args, in RecvAsync()
177 const Rendezvous::Args& args, in Send()
185 const Rendezvous::Args& args, in RecvAsync()
Dgraph_runner.cc134 string full_key = Rendezvous::CreateKey("/device:CPU:0", 1, "/device:CPU:1", in Run()
136 Rendezvous::ParsedKey parsed; in Run()
137 TF_RETURN_IF_ERROR(Rendezvous::ParseKey(full_key, &parsed)); in Run()
138 TF_RETURN_IF_ERROR(rendez.Send(parsed, Rendezvous::Args(), in.second, in Run()
167 Rendezvous** r) { in Run()
196 Rendezvous::CreateKey("/device:CPU:0", 1, "/device:CPU:1", in Run()
198 Rendezvous::ParsedKey parsed; in Run()
199 TF_RETURN_IF_ERROR(Rendezvous::ParseKey(output_key, &parsed)); in Run()
203 rendez.Recv(parsed, Rendezvous::Args(), &output_tensor, &is_dead)); in Run()
Dexecutor_test.cc74 Rendezvous** r) { in Create()
84 Status Run(Rendezvous* rendez) { in Run()
98 Rendezvous* rendez_ = nullptr;
138 Rendezvous::ParsedKey Key(const string& sender, const uint64 incarnation, in Key()
140 Rendezvous::ParsedKey result; in Key()
142 Rendezvous::ParseKey(Rendezvous::CreateKey(sender, incarnation, receiver, in Key()
160 Rendezvous::Args args; in TEST_F()
191 Rendezvous::Args args; in TEST_F()
242 Rendezvous::Args args; in TEST_F()
276 Rendezvous* rendez = NewLocalRendezvous(); in TEST_F()
[all …]
Drendezvous_mgr.h43 class RefCountedIntraProcessRendezvous : public Rendezvous {
48 Status Send(const ParsedKey& key, const Rendezvous::Args& args,
50 void RecvAsync(const ParsedKey& key, const Rendezvous::Args& args,
77 Status Send(const ParsedKey& key, const Rendezvous::Args& args,
79 void RecvAsync(const ParsedKey& key, const Rendezvous::Args& args,
Dexecutor.h86 typedef std::function<Status(const int64, const DeviceMgr*, Rendezvous** r)>
173 ExecutorBarrier(size_t num, Rendezvous* r, StatusCallback done) in ExecutorBarrier()
185 Rendezvous* rendez_ = nullptr;
193 Rendezvous* error_rendez = nullptr; in WhenDone()
/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()
136 OP_REQUIRES_OK(ctx, Rendezvous::ParseKey(parsed_key_.buf_, &parsed_key_)); in RecvOp()
143 Rendezvous::DoneCallback make_recv_callback(OpKernelContext* ctx, in make_recv_callback()
146 const Rendezvous::Args& send_args, in make_recv_callback()
147 const Rendezvous::Args& recv_args, in make_recv_callback()
169 Rendezvous::Args args; in ComputeAsync()
185 Rendezvous::ParsedKey in_loop_parsed; in ComputeAsync()
[all …]
Dsendrecv_ops.h31 Rendezvous::ParsedKey parsed_key_;
44 Rendezvous::ParsedKey parsed_key_;
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcollective_permute_thunk.cc108 class Rendezvous { class
110 explicit Rendezvous(const RendezvousKey& key) : key_(key) {} in Rendezvous() function in xla::gpu::__anone743f6410111::Rendezvous
158 StatusOr<std::shared_ptr<BlockingCounter>> Rendezvous::SubmitParticipant( in SubmitParticipant()
213 RefcountingHashMap<RendezvousKey, Rendezvous>& GlobalRendezvousMap() { in GlobalRendezvousMap()
214 static auto& m = *new RefcountingHashMap<RendezvousKey, Rendezvous>( in GlobalRendezvousMap()
216 return absl::make_unique<Rendezvous>(key); in GlobalRendezvousMap()
236 std::shared_ptr<Rendezvous> rendezvous = GlobalRendezvousMap()[key]; in ExecuteOnStream()
/external/tensorflow/tensorflow/c/experimental/
Dnetwork_test.cc156 Rendezvous::ParsedKey Key(const string& sender, const uint64 incarnation, in Key()
158 Rendezvous::ParsedKey result; in Key()
160 Rendezvous::ParseKey(Rendezvous::CreateKey(sender, incarnation, receiver, in Key()
231 Rendezvous::Args args; in TEST_F()
241 const Status&, const Rendezvous::Args&, const Rendezvous::Args&, in TEST_F()
Drendezvous.cc45 void CRemoteRendezvous::RecvFromRemoteAsync(const Rendezvous::ParsedKey& parsed, in RecvFromRemoteAsync()
46 const Rendezvous::Args& args, in RecvFromRemoteAsync()
108 ::tensorflow::Rendezvous::Args recv_args; in TF_RendezvousDone()
112 ::tensorflow::Rendezvous::Args sent_args; in TF_RendezvousDone()
Drendezvous_internal.h61 ::tensorflow::Rendezvous::DoneCallback done_callback;
93 void RecvFromRemoteAsync(const Rendezvous::ParsedKey& parsed,
94 const Rendezvous::Args& args,
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.h126 Rendezvous* rendezvous,
227 Rendezvous* GetRendezvous() const { return rendezvous_; } in GetRendezvous()
228 Rendezvous* CreateRendezvous(const int64 step_id) const { in CreateRendezvous()
321 Rendezvous* r, DeviceMgr* local_device_mgr, int keep_alive_secs,
337 Rendezvous* r, DeviceMgr* local_device_mgr, int keep_alive_secs,
347 std::function<Rendezvous*(const int64)> rendezvous_creator,
493 Rendezvous* rendezvous_;
494 std::function<Rendezvous*(const int64)> rendezvous_creator_;
560 uint64 context_id, uint64 context_view_id, Rendezvous* r,
/external/tensorflow/tensorflow/compiler/xla/service/
Dcollective_ops_utils.h187 class Rendezvous {
189 virtual ~Rendezvous() {} in ~Rendezvous()
190 explicit Rendezvous(const RendezvousKey& k) : key_(k) {} in Rendezvous() function
195 std::function<std::shared_ptr<Rendezvous<O>>()> rendezvous_getter, in SubmitParticipant()
197 std::shared_ptr<Rendezvous<O>> rendezvous = rendezvous_getter(); in SubmitParticipant()

123