| /external/tensorflow/tensorflow/core/common_runtime/ |
| rendezvous_mgr.h | 23 #include "tensorflow/core/framework/rendezvous.h"
32 // IntraProcessRendezvous is a Rendezvous which expects all producers
37 // Buffering of Tensor values is delegated to a "local" Rendezvous
40 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.
70 // rendezvous key's source is in this process. If "is_src" is false,
71 // checks that the rendezvous key's destination is in this process.
73 Rendezvous::ParsedKey* parsed)
[all...] |
| rendezvous_util.cc | 20 Rendezvous* rendezvous, DeviceContext* device_context,
35 if (!rendezvous) {
36 return errors::InvalidArgument("Rendezvous is null.");
39 Rendezvous::ParsedKey parsed;
41 Rendezvous::Args rendez_args;
46 TF_RETURN_IF_ERROR(Rendezvous::ParseKey(keys[i], &parsed));
48 rendezvous->Send(parsed, rendez_args, tensors_to_send[i], false));
54 Rendezvous* rendezvous, DeviceContext* device_context
[all...] |
| rendezvous_util.h | 20 #include "tensorflow/core/framework/rendezvous.h"
27 // Uses `rendezvous` to send tensors in `tensors_to_send`. `device_context`
33 Rendezvous* rendezvous, DeviceContext* device_context,
37 // Uses `rendezvous` to obtain tensors. `device_context` should be the
42 Rendezvous* rendezvous, DeviceContext* device_context,
47 Status RecvOutputsFromRendezvous(Rendezvous* rendezvous, NamedTensors* out,
48 const Rendezvous::Args& args)
[all...] |
| kernel_benchmark_testlib.h | 42 Rendezvous* rendez = nullptr);
59 Rendezvous* rendez_ = nullptr;
|
| rendezvous_mgr.cc | 40 const Rendezvous::Args& args,
53 Rendezvous::ParsedKey* parsed) {
58 TF_RETURN_IF_ERROR(Rendezvous::ParseKey(key, parsed));
63 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& send_args,
64 const Rendezvous::Args& recv_args, const Tensor& in, Tensor* out,
117 const Rendezvous::Args& recv_args,
125 const Status& status, const Rendezvous::Args& send_args,
126 const Rendezvous::Args& recv_args, const Tensor& in, bool is_dead) {
|
| kernel_benchmark_testlib.cc | 45 Rendezvous* rendez) {
92 args.rendezvous = rendez_;
122 return Rendezvous::CreateKey(send_device, send_device_incarnation,
132 // Gets inputs' and outputs' rendezvous keys.
148 args.rendezvous = rendez_;
155 Rendezvous::ParsedKey parsed;
156 TF_CHECK_OK(Rendezvous::ParseKey(p.first, &parsed));
157 TF_CHECK_OK(rendez_->Send(parsed, Rendezvous::Args(), p.second, false));
161 Rendezvous::ParsedKey parsed;
162 TF_CHECK_OK(Rendezvous::ParseKey(key, &parsed))
[all...] |
| executor.h | 20 #include "tensorflow/core/framework/rendezvous.h"
39 // Rendezvous* rendezvous = NewNaiveRendezvous();
40 // TF_CHECK_OK(rendezvous->Send("input", some_input_tensor));
41 // TF_CHECK_OK(executor->Run({ExecutorOpts, rendezvous, nullptr}));
42 // TF_CHECK_OK(rendezvous->Recv("output", &output_tensor));
55 // ensure objects passed in Args (rendezvous, stats_collector, etc.)
64 // RunAsync() uses the given "rendezvous", if not null, as the
85 Rendezvous* rendezvous = nullptr member in struct:tensorflow::Executor::Args
[all...] |
| graph_runner.cc | 51 // A simple rendezvous class.
54 class SimpleRendezvous : public Rendezvous {
137 string full_key = Rendezvous::CreateKey("/device:CPU:0", 1, "/device:CPU:1",
139 Rendezvous::ParsedKey parsed;
140 TF_RETURN_IF_ERROR(Rendezvous::ParseKey(full_key, &parsed));
141 TF_RETURN_IF_ERROR(rendez->Send(parsed, Rendezvous::Args(), in.second,
152 // Create the local executor and the Rendezvous for fetching back the
182 args.rendezvous = rendez;
190 Rendezvous::CreateKey("/device:CPU:0", 1, "/device:CPU:1",
192 Rendezvous::ParsedKey parsed
[all...] |
| rendezvous_util_test.cc | 30 Rendezvous* rendez_;
48 return Rendezvous::CreateKey(
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| rendezvous_mgr_interface.h | 22 #include "tensorflow/core/framework/rendezvous.h"
35 // Partially initialized RemoteRendezvous must respect the Rendezvous interface
39 class RemoteRendezvous : public Rendezvous {
45 // RendezvousMgr keeps track of a set of local rendezvous instances.
48 // corresponds to one local rendezvous instance managed by a
52 // Rendezvous* rendez = worker_env->rendezvous_mgr->Find(0x8935);
61 // Tensors sent and recved through rendezvous managed by this
62 // RendezvousMgr must have keys generated by Rendezvous::CreateKey.
68 // Returns Rendezvous supporting send and recv among workers in the
70 // returned Rendezvous instance
[all...] |
| base_rendezvous_mgr.h | 26 #include "tensorflow/core/framework/rendezvous.h"
42 // RendezvousMgr keeps track of a set of local rendezvous instances.
45 // corresponds to one local rendezvous instance managed by a
49 // Rendezvous* rendez = worker_env->rendezvous_mgr->Find(0x8935);
58 // Tensors sent and received through a rendezvous managed by this
59 // RendezvousMgr must have keys generated by Rendezvous::CreateKey().
66 // Returns Rendezvous supporting send and recv among workers in the
68 // returned Rendezvous instance.
74 // Finds the local rendezvous instance for the "step_id". Runs
78 void RecvLocalAsync(int64 step_id, const Rendezvous::ParsedKey& parsed
[all...] |
| base_rendezvous_mgr.cc | 38 static void StartAbortRendevous(Rendezvous* rendez, const Status& s) {
69 const Rendezvous::ParsedKey& parsed,
70 Rendezvous::DoneCallback done) {
73 Rendezvous::DoneCallback done_cb = std::bind(
74 [rendez](Rendezvous::DoneCallback done,
76 const Status& s, const Rendezvous::Args& send_args,
77 const Rendezvous::Args& recv_args, const Tensor& v, bool dead) {
86 const Rendezvous::ParsedKey& parsed,
92 const Rendezvous::Args& send_args,
93 const Rendezvous::Args& recv_args
[all...] |
| executor_test.cc | 25 #include "tensorflow/core/framework/rendezvous.h"
52 // There should always be exactly one Ref left on the Rendezvous
77 Status Run(Rendezvous* rendez) {
79 args.rendezvous = rendez;
91 Rendezvous* rendez_ = nullptr;
131 Rendezvous::ParsedKey Key(const string& sender, const uint64 incarnation,
133 Rendezvous::ParsedKey result;
135 Rendezvous::ParseKey(Rendezvous::CreateKey(sender, incarnation, receiver,
153 Rendezvous::Args args
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| sendrecv_ops.h | 31 Rendezvous::ParsedKey parsed_key_;
44 Rendezvous::ParsedKey parsed_key_;
|
| sendrecv_ops.cc | 48 // to formulate the unique rendezvous key.
69 // proactively cache the rendezvous key for the top-level.
71 OP_REQUIRES_OK(ctx, Rendezvous::ParseKey(parsed_key_.buf_, &parsed_key_));
79 ctx, ctx->rendezvous() != nullptr,
80 errors::Internal("Op kernel context needs to provide a rendezvous."));
86 Rendezvous::Args args;
92 // Use the cached rendezvous key.
94 ctx->SetStatus(ctx->rendezvous()->Send(parsed_key_, args, ctx->input(0),
98 Rendezvous::ParsedKey in_loop_parsed;
102 Rendezvous::ParseKey(in_loop_parsed.buf_, &in_loop_parsed))
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| rendezvous_test.cc | 16 #include "tensorflow/core/framework/rendezvous.h"
40 const string key = Rendezvous::CreateKey(
49 Rendezvous::ParsedKey parsed;
50 TF_EXPECT_OK(Rendezvous::ParseKey(key, &parsed));
57 EXPECT_FALSE(Rendezvous::ParseKey("foo;bar;baz", &parsed).ok());
58 EXPECT_FALSE(Rendezvous::ParseKey("/job:mnist/replica:1/task:2/CPU:0;"
63 Rendezvous::ParseKey(strings::StrCat(key, ";", key), &parsed).ok());
78 Rendezvous* rendez_;
98 Rendezvous::ParsedKey MakeKey(const string& name) {
99 string s = Rendezvous::CreateKey("/job:mnist/replica:1/task:2/CPU:0", 7890
[all...] |
| rendezvous.h | 30 // A Rendezvous is an abstraction for passing tensors from producers
31 // to consumers. A rendezvous is a table of channels. Each channel is
32 // keyed by a rendezvous key. The key encodes a pair of <producer,
46 class Rendezvous : public core::RefCounted {
53 // Constructs a rendezvous key for the tensor of "name" sent from
77 friend class Rendezvous;
98 // Callback provided by a tensor consumer waiting on the rendezvous.
101 // two Rendezvous::Args, one provided by the sender, the other by the
124 ~Rendezvous() override;
127 // Returns a Rendezvous instance that is limited to use only b
[all...] |
| rendezvous.cc | 16 #include "tensorflow/core/framework/rendezvous.h"
36 Rendezvous::ParsedKey& Rendezvous::ParsedKey::operator=(const ParsedKey& b) {
52 string Rendezvous::CreateKey(const string& src_device, uint64 src_incarnation,
86 Status Rendezvous::ParseKey(StringPiece key, ParsedKey* out) {
112 return errors::InvalidArgument("Invalid rendezvous key: ", key);
115 Rendezvous::~Rendezvous() {}
117 Status Rendezvous::Recv(const ParsedKey& key, const Args& recv_args,
143 Status Rendezvous::Recv(const ParsedKey& key, const Args& args, Tensor* val
[all...] |
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| rpc_rendezvous_mgr_test.cc | 43 Rendezvous::ParsedKey MakeKey(const string& s) {
44 Rendezvous::ParsedKey key;
45 CHECK(Rendezvous::ParseKey(s, &key).ok());
86 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey(
93 Rendezvous::Args args;
106 const Rendezvous::ParsedKey key = MakeKey(Rendezvous::CreateKey(
119 Rendezvous::Args args;
133 Rendezvous::Args args
[all...] |
| rpc_rendezvous_mgr.cc | 46 void RecvFromRemoteAsync(const Rendezvous::ParsedKey& parsed,
47 const Rendezvous::Args& args,
63 const Rendezvous::Args& recv_args, Rendezvous::DoneCallback done) {
121 const Rendezvous::Args& recv_args() const { return recv_args_; }
122 const Rendezvous::DoneCallback& done() const { return done_; }
153 Rendezvous::Args recv_args_;
154 Rendezvous::DoneCallback done_;
208 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& recv_args
[all...] |
| /external/tensorflow/tensorflow/contrib/verbs/ |
| rdma_rendezvous_mgr.cc | 36 void RecvFromRemoteAsync(const Rendezvous::ParsedKey& parsed,
37 const Rendezvous::Args& args,
48 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& recv_args,
|
| /external/tensorflow/tensorflow/contrib/mpi/ |
| mpi_rendezvous_mgr.h | 72 void Init(const Rendezvous::ParsedKey& parsed, const int64 step_id,
92 Rendezvous::DoneCallback done_;
106 void Init(const Rendezvous::ParsedKey& parsed, const int64 step_id) {
125 void RecvFromRemoteAsync(const Rendezvous::ParsedKey& parsed,
126 const Rendezvous::Args& args,
163 const Status&, const Rendezvous::Args&, const Rendezvous::Args&,
|
| mpi_rendezvous_mgr.cc | 63 const Rendezvous::ParsedKey& parsed, const Rendezvous::Args& recv_args,
157 Rendezvous::ParsedKey parsed;
158 TF_CHECK_OK(Rendezvous::ParseKey(key, &parsed));
162 const Rendezvous::Args& send_args,
163 const Rendezvous::Args& recv_args,
198 Rendezvous::DoneCallback done_cb =
200 const Status& status, const Rendezvous::Args& send_args,
201 const Rendezvous::Args& recv_args, const Tensor& val, bool is_dead) {
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| gdr_worker.cc | 62 Rendezvous::ParsedKey parsed;
63 s = Rendezvous::ParseKey(key, &parsed);
73 // Request the tensor associated with the rendezvous key. Any time
76 // cancellation should abort the rendezvous.
82 const Status& status, const Rendezvous::Args& send_args,
83 const Rendezvous::Args&, const Tensor& val, const bool is_dead) {
|
| gdr_rendezvous_mgr.cc | 43 const Rendezvous::Args& recv_args, int64 step_id,
109 const Rendezvous::Args& recv_args() const { return recv_args_; }
118 Rendezvous::Args recv_args_;
134 void RecvFromRemoteAsync(const Rendezvous::ParsedKey& parsed,
135 const Rendezvous::Args& recv_args,
|
