| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous_test.cc | 42 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 …]
|
| D | local_rendezvous.cc | 36 Item(Rendezvous::Args send_args, const Tensor& value, bool is_dead) in Item()
42 Item(Rendezvous::Args recv_args, Rendezvous::DoneCallback waiter, in Item()
60 const Rendezvous::Args args;
74 ManualConstructor<Rendezvous::DoneCallback> waiter;
80 Item(Rendezvous::Args args, Type type) : args(args), type(type) { in Item()
109 Status LocalRendezvous::Send(const Rendezvous::ParsedKey& key, in Send()
110 const Rendezvous::Args& send_args, in Send()
167 void LocalRendezvous::RecvAsync(const Rendezvous::ParsedKey& key, in RecvAsync()
168 const Rendezvous::Args& recv_args, in RecvAsync()
169 Rendezvous::DoneCallback done) { in RecvAsync()
[all …]
|
| D | local_rendezvous.h | 42 explicit LocalRendezvous(Rendezvous* owner) : rc_owner_(owner) {} in LocalRendezvous()
45 Status Send(const Rendezvous::ParsedKey& key,
46 const Rendezvous::Args& send_args, const Tensor& val,
48 void RecvAsync(const Rendezvous::ParsedKey& key,
49 const Rendezvous::Args& recv_args,
50 Rendezvous::DoneCallback done);
70 const Rendezvous* rc_owner_;
|
| D | rendezvous.h | 74 friend class Rendezvous;
130 class Rendezvous : public RendezvousInterface, public core::RefCounted {
137 Factory(std::function<Status(const int64, const DeviceMgr*, Rendezvous**)> in Factory()
147 std::function<Status(const int64, const DeviceMgr*, Rendezvous**)> in Factory()
156 Rendezvous** rendez) const { in operator()
164 std::function<Status(const int64, const DeviceMgr*, Rendezvous**)>
184 Rendezvous* NewLocalRendezvous();
|
| D | rendezvous.cc | 38 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()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | kernel_benchmark_testlib.cc | 51 Rendezvous* rendez, const char* executor_type, in Benchmark()
96 Rendezvous::Factory())); in Benchmark()
157 return Rendezvous::CreateKey(send_device, send_device_incarnation, in GetRendezvousKey()
181 Rendezvous::ParsedKey parsed; in RunWithRendezvousArgs()
182 TF_CHECK_OK(Rendezvous::ParseKey(p.first, &parsed)); in RunWithRendezvousArgs()
183 TF_CHECK_OK(rendez_->Send(parsed, Rendezvous::Args(), p.second, false)); in RunWithRendezvousArgs()
187 Rendezvous::ParsedKey parsed; in RunWithRendezvousArgs()
188 TF_CHECK_OK(Rendezvous::ParseKey(key, &parsed)); in RunWithRendezvousArgs()
189 TF_CHECK_OK(rendez_->Recv(parsed, Rendezvous::Args(), &unused, &is_dead)); in RunWithRendezvousArgs()
199 Rendezvous::ParsedKey parsed; in RunWithRendezvousArgs()
[all …]
|
| D | rendezvous_util.cc | 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()
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 …]
|
| D | rendezvous_mgr.cc | 37 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()
118 const Rendezvous::Args& recv_args, in IntraProcessRecvAsyncImpl()
127 const Status& status, const Rendezvous::Args& send_args, in IntraProcessRecvAsyncImpl()
128 const Rendezvous::Args& recv_args, const Tensor& in, in IntraProcessRecvAsyncImpl()
159 const Rendezvous::Args& args, in Send()
167 const Rendezvous::Args& args, in RecvAsync()
184 const Rendezvous::Args& args, in Send()
192 const Rendezvous::Args& args, in RecvAsync()
|
| D | rendezvous_mgr.h | 43 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,
|
| D | graph_runner.cc | 136 string full_key = Rendezvous::CreateKey("/device:CPU:0", 1, "/device:CPU:1", in Run()
138 Rendezvous::ParsedKey parsed; in Run()
139 TF_RETURN_IF_ERROR(Rendezvous::ParseKey(full_key, &parsed)); in Run()
140 TF_RETURN_IF_ERROR(rendez.Send(parsed, Rendezvous::Args(), in.second, in Run()
194 Rendezvous::CreateKey("/device:CPU:0", 1, "/device:CPU:1", in Run()
196 Rendezvous::ParsedKey parsed; in Run()
197 TF_RETURN_IF_ERROR(Rendezvous::ParseKey(output_key, &parsed)); in Run()
201 rendez.Recv(parsed, Rendezvous::Args(), &output_tensor, &is_dead)); in Run()
|
| D | executor_test.cc | 89 Status Run(Rendezvous* rendez) { in Run()
103 Rendezvous* rendez_ = nullptr;
143 Rendezvous::ParsedKey Key(const string& sender, const uint64 incarnation, in Key()
145 Rendezvous::ParsedKey result; in Key()
147 Rendezvous::ParseKey(Rendezvous::CreateKey(sender, incarnation, receiver, in Key()
165 Rendezvous::Args args; in TEST_F()
196 Rendezvous::Args args; in TEST_F()
247 Rendezvous::Args args; in TEST_F()
281 Rendezvous* rendez = NewLocalRendezvous(); in TEST_F()
283 Rendezvous::Args args; in TEST_F()
[all …]
|
| D | executor.h | 163 ExecutorBarrier(size_t num, Rendezvous* r, StatusCallback done) in ExecutorBarrier()
175 Rendezvous* rendez_ = nullptr;
183 Rendezvous* error_rendez = nullptr; in WhenDone()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr_test.cc | 47 Rendezvous::ParsedKey MakeKey(const string& s) { in MakeKey()
48 Rendezvous::ParsedKey key; in MakeKey()
49 CHECK(Rendezvous::ParseKey(s, &key).ok()); in MakeKey()
141 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
148 Rendezvous::Args args; in TEST_F()
161 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
174 Rendezvous::Args args; in TEST_F()
188 Rendezvous::Args args; in TEST_F()
195 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey( in TEST_F()
210 Rendezvous::Args args; in TEST_F()
[all …]
|
| D | rpc_rendezvous_mgr.cc | 47 void RecvFromRemoteAsync(const Rendezvous::ParsedKey& parsed,
48 const Rendezvous::Args& args,
64 const Rendezvous::Args& recv_args, Rendezvous::DoneCallback done) { in Init()
131 const Rendezvous::Args& recv_args() const { return recv_args_; } in recv_args()
132 const Rendezvous::DoneCallback& done() const { return done_; } in done()
179 Rendezvous::Args recv_args_;
180 Rendezvous::DoneCallback done_;
234 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& recv_args, in RecvFromRemoteAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | base_rendezvous_mgr.h | 78 void RecvLocalAsync(int64 step_id, const Rendezvous::ParsedKey& parsed,
79 Rendezvous::DoneCallback done) override;
82 Status RecvLocal(int64 step_id, const Rendezvous::ParsedKey& parsed,
125 Status Send(const ParsedKey& key, const Rendezvous::Args& args,
132 void RecvAsync(const ParsedKey& key, const Rendezvous::Args& args,
150 virtual void RecvFromRemoteAsync(const Rendezvous::ParsedKey& parsed,
151 const Rendezvous::Args& args,
160 void RegisterCall(BaseRecvTensorCall* call, const Rendezvous::Args& args);
175 Rendezvous* local_; // Owns a Ref on this object.
205 Status ValidateDevices(const Rendezvous::ParsedKey& parsed, bool is_src);
[all …]
|
| D | base_rendezvous_mgr.cc | 41 static void StartAbortRendevous(Rendezvous* rendez, const Status& s) { in StartAbortRendevous()
72 const Rendezvous::ParsedKey& parsed, in RecvLocalAsync()
73 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, 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()
105 Rendezvous* rendez = nullptr; in Cleanup()
173 Status BaseRemoteRendezvous::Send(const Rendezvous::ParsedKey& parsed, in Send()
[all …]
|
| D | rendezvous_mgr_interface.h | 39 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/kernels/ |
| D | sendrecv_ops.cc | 74 OP_REQUIRES_OK(ctx, Rendezvous::ParseKey(parsed_key_.buf_, &parsed_key_)); in SendOp()
89 Rendezvous::Args args; in Compute()
102 Rendezvous::ParsedKey in_loop_parsed; in Compute()
107 Rendezvous::ParseKey(in_loop_parsed.buf_, &in_loop_parsed)); in Compute()
153 OP_REQUIRES_OK(ctx, Rendezvous::ParseKey(parsed_key_.buf_, &parsed_key_)); in RecvOp()
170 Rendezvous::DoneCallback make_recv_callback(OpKernelContext* ctx, in make_recv_callback()
173 const Rendezvous::Args& send_args, in make_recv_callback()
174 const Rendezvous::Args& recv_args, in make_recv_callback()
196 Rendezvous::Args args; in ComputeAsync()
208 Rendezvous::ParsedKey in_loop_parsed; in ComputeAsync()
[all …]
|
| D | sendrecv_ops.h | 33 Rendezvous::ParsedKey parsed_key_;
48 Rendezvous::ParsedKey parsed_key_;
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | host_compute_ops.cc | 126 std::vector<Rendezvous::ParsedKey> parsed_key(ctx->num_outputs()); in ComputeAsync()
128 rendezvous_key[i] = Rendezvous::CreateKey( in ComputeAsync()
135 ctx, Rendezvous::ParseKey(rendezvous_key[i], &parsed_key[i]), done); in ComputeAsync()
143 Rendezvous::Args args; in ComputeAsync()
152 const Rendezvous::Args& send_args, in ComputeAsync()
153 const Rendezvous::Args& recv_args, in ComputeAsync()
270 Rendezvous::Args args; in Compute()
275 const string& rendezvous_key = Rendezvous::CreateKey( in Compute()
281 Rendezvous::ParsedKey parsed_key; in Compute()
282 OP_REQUIRES_OK(ctx, Rendezvous::ParseKey(rendezvous_key, &parsed_key)); in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | collective_permute_thunk.cc | 109 class Rendezvous { class
111 explicit Rendezvous(const RendezvousKey& key) : key_(key) {} in Rendezvous() function in xla::gpu::__anonb6c88d1d0111::Rendezvous
159 StatusOr<std::shared_ptr<BlockingCounter>> Rendezvous::SubmitParticipant( in SubmitParticipant()
214 RefcountingHashMap<RendezvousKey, Rendezvous>& GlobalRendezvousMap() { in GlobalRendezvousMap()
215 static auto& m = *new RefcountingHashMap<RendezvousKey, Rendezvous>(); in GlobalRendezvousMap()
238 return absl::make_unique<Rendezvous>(key); in ExecuteOnStream()
240 std::shared_ptr<Rendezvous> rendezvous = in ExecuteOnStream()
|
| /external/llvm-project/lldb/source/Plugins/DynamicLoader/Hexagon-DYLD/ |
| D | HexagonDYLDRendezvous.h | 37 struct Rendezvous { struct
44 Rendezvous() in Rendezvous() argument
193 Rendezvous m_current;
194 Rendezvous m_previous;
|
| /external/llvm-project/lldb/source/Plugins/DynamicLoader/POSIX-DYLD/ |
| D | DYLDRendezvous.h | 40 struct Rendezvous { struct
47 Rendezvous() : version(0), map_addr(0), brk(0), state(0), ldbase(0) {} in Rendezvous() argument
184 Rendezvous m_current;
185 Rendezvous m_previous;
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | collective_ops_utils.h | 209 class Rendezvous {
211 virtual ~Rendezvous() {} in ~Rendezvous()
212 explicit Rendezvous(const RendezvousKey& k) : key_(k) {} in Rendezvous() function
217 std::function<std::shared_ptr<Rendezvous<I, O>>()> rendezvous_getter, in SubmitParticipant()
219 std::shared_ptr<Rendezvous<I, O>> rendezvous = rendezvous_getter(); in SubmitParticipant()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.h | 101 Rendezvous* rendezvous,
271 Rendezvous* GetRendezvous() const { return rendezvous_; } in GetRendezvous()
280 std::function<Rendezvous*(int64)> RendezvousCreator() { in RendezvousCreator()
371 Rendezvous* r, const DeviceMgr* local_device_mgr, int keep_alive_secs,
395 std::function<Rendezvous*(const int64)> rendezvous_creator,
500 Rendezvous* CreateRendezvous(int64 step_id) const { in CreateRendezvous()
592 Rendezvous* rendezvous_;
593 std::function<Rendezvous*(const int64)> rendezvous_creator_;
672 uint64 context_id, uint64 context_view_id, Rendezvous* r,
|