| /external/tensorflow/tensorflow/core/common_runtime/ |
| rendezvous_util_test.cc | 26 RendezvousUtilTest() { rendez_ = NewLocalRendezvous(); }
28 ~RendezvousUtilTest() override { rendez_->Unref(); }
30 Rendezvous* rendez_; member in class:tensorflow::__anon39505::RendezvousUtilTest
56 rendez_, nullptr, {}, {MakeStringKey("hello1"), MakeStringKey("hello2")},
62 rendez_, nullptr, {}, {MakeStringKey("hello1"), MakeStringKey("hello2")},
76 rendez_, nullptr, {}, {MakeStringKey("hello1"), MakeStringKey("hello2")},
80 rendez_, nullptr, {}, {MakeStringKey("hello1"), MakeStringKey("hello2")},
|
| kernel_benchmark_testlib.cc | 68 rendez_ = NewLocalRendezvous();
70 rendez_ = rendez;
92 args.rendezvous = rendez_;
103 rendez_->Unref();
148 args.rendezvous = rendez_;
157 TF_CHECK_OK(rendez_->Send(parsed, Rendezvous::Args(), p.second, false));
163 TF_CHECK_OK(rendez_->Recv(parsed, Rendezvous::Args(), &unused, &is_dead));
174 TF_CHECK_OK(rendez_->Send(parsed, Rendezvous::Args(), p.second, false));
180 TF_CHECK_OK(rendez_->Recv(parsed, Rendezvous::Args(), &unused, &is_dead));
|
| kernel_benchmark_testlib.h | 59 Rendezvous* rendez_ = nullptr; member in class:tensorflow::test::Benchmark
|
| executor.h | 166 : rendez_(r), done_cb_(done), pending_(num) {}
177 Rendezvous* rendez_ = nullptr; member in class:tensorflow::ExecutorBarrier
196 error_rendez = rendez_;
|
| /external/tensorflow/tensorflow/core/framework/ |
| rendezvous_test.cc | 69 rendez_ = NewLocalRendezvous();
72 ~LocalRendezvousTest() override { rendez_->Unref(); }
78 Rendezvous* rendez_; member in class:tensorflow::__anon39590::LocalRendezvousTest
119 TF_ASSERT_OK(rendez_->Send(KeyFoo(), args, V("hello"), false));
122 TF_ASSERT_OK(rendez_->Recv(KeyFoo(), args, &val, &is_dead));
130 TF_ASSERT_OK(rendez_->Send(KeyFoo(), args, V("hello"), false));
135 TF_ASSERT_OK(rendez_->Recv(KeyFoo(), args, &val, &is_dead));
144 TF_ASSERT_OK(rendez_->Recv(KeyFoo(), args, &t, &is_dead));
145 TF_ASSERT_OK(rendez_->Send(KeyBar(), args, t, is_dead));
151 TF_ASSERT_OK(rendez_->Send(KeyFoo(), args, V("secret msg"), val_dead))
[all...] |
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| executor_test.cc | 54 CHECK(rendez_->Unref());
74 rendez_ = NewLocalRendezvous();
91 Rendezvous* rendez_ = nullptr; member in class:tensorflow::ExecutorTest
154 TF_ASSERT_OK(rendez_->Send(Key(ALICE, kIncarnation, BOB, "a"), args, V(1.0),
156 TF_ASSERT_OK(rendez_->Send(Key(ALICE, kIncarnation, BOB, "b"), args, V(1.0),
158 TF_ASSERT_OK(Run(rendez_));
162 rendez_->Recv(Key(BOB, kIncarnation, ALICE, "c"), args, &out, &is_dead));
187 rendez_->Send(Key(ALICE, kIncarnation, BOB, "a"), args, V(1.0), false));
188 TF_ASSERT_OK(Run(rendez_));
192 rendez_->Recv(Key(BOB, kIncarnation, ALICE, "b"), args, &out, &is_dead))
[all...] |
| /external/tensorflow/tensorflow/c/eager/ |
| runtime.h | 175 : device_(nullptr), flib_(nullptr), rendez_(rendez) {}
188 Rendezvous* rendez_; member in class:tensorflow::KernelAndDevice
|
| runtime.cc | 286 params.rendezvous = rendez_;
|
