| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous_test.cc | 71 rendez_ = NewLocalRendezvous(); in LocalRendezvousTest()
74 ~LocalRendezvousTest() override { rendez_->Unref(); } in ~LocalRendezvousTest()
80 Rendezvous* rendez_; member in tensorflow::__anon404285390111::LocalRendezvousTest
121 TF_ASSERT_OK(rendez_->Send(KeyFoo(), args, V("hello"), false)); in TEST_F()
124 TF_ASSERT_OK(rendez_->Recv(KeyFoo(), args, &val, &is_dead)); in TEST_F()
132 TF_ASSERT_OK(rendez_->Send(KeyFoo(), args, V("hello"), false)); in TEST_F()
137 TF_ASSERT_OK(rendez_->Recv(KeyFoo(), args, &val, &is_dead)); in TEST_F()
146 TF_ASSERT_OK(rendez_->Recv(KeyFoo(), args, &t, &is_dead)); in TEST_F()
147 TF_ASSERT_OK(rendez_->Send(KeyBar(), args, t, is_dead)); in TEST_F()
153 TF_ASSERT_OK(rendez_->Send(KeyFoo(), args, V("secret msg"), val_dead)); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | executor_test.cc | 65 CHECK(rendez_->Unref()); in ~ExecutorTest()
83 rendez_ = NewLocalRendezvous(); in Create()
103 Rendezvous* rendez_ = nullptr; member in tensorflow::ExecutorTest
166 TF_ASSERT_OK(rendez_->Send(Key(ALICE, kIncarnation, BOB, "a"), args, V(1.0), in TEST_F()
168 TF_ASSERT_OK(rendez_->Send(Key(ALICE, kIncarnation, BOB, "b"), args, V(1.0), in TEST_F()
170 TF_ASSERT_OK(Run(rendez_)); in TEST_F()
174 rendez_->Recv(Key(BOB, kIncarnation, ALICE, "c"), args, &out, &is_dead)); in TEST_F()
199 rendez_->Send(Key(ALICE, kIncarnation, BOB, "a"), args, V(1.0), false)); in TEST_F()
200 TF_ASSERT_OK(Run(rendez_)); in TEST_F()
204 rendez_->Recv(Key(BOB, kIncarnation, ALICE, "b"), args, &out, &is_dead)); in TEST_F()
[all …]
|
| D | rendezvous_util_test.cc | 26 RendezvousUtilTest() { rendez_ = NewLocalRendezvous(); } in RendezvousUtilTest()
28 ~RendezvousUtilTest() override { rendez_->Unref(); } in ~RendezvousUtilTest()
30 Rendezvous* rendez_; member in tensorflow::__anon6a2846f40111::RendezvousUtilTest
56 rendez_, nullptr, {}, {MakeStringKey("hello1"), MakeStringKey("hello2")}, in TEST_F()
62 rendez_, nullptr, {}, {MakeStringKey("hello1"), MakeStringKey("hello2")}, in TEST_F()
76 rendez_, nullptr, {}, {MakeStringKey("hello1"), MakeStringKey("hello2")}, in TEST_F()
80 rendez_, nullptr, {}, {MakeStringKey("hello1"), MakeStringKey("hello2")}, in TEST_F()
|
| D | kernel_benchmark_testlib.cc | 83 rendez_ = NewLocalRendezvous(); in Benchmark()
85 rendez_ = rendez; in Benchmark()
117 args.rendezvous = rendez_; in Benchmark()
130 rendez_->Unref(); in ~Benchmark()
174 args.rendezvous = rendez_; in RunWithRendezvousArgs()
183 TF_CHECK_OK(rendez_->Send(parsed, Rendezvous::Args(), p.second, false)); in RunWithRendezvousArgs()
189 TF_CHECK_OK(rendez_->Recv(parsed, Rendezvous::Args(), &unused, &is_dead)); in RunWithRendezvousArgs()
201 TF_CHECK_OK(rendez_->Send(parsed, Rendezvous::Args(), p.second, false)); in RunWithRendezvousArgs()
207 TF_CHECK_OK(rendez_->Recv(parsed, Rendezvous::Args(), &unused, &is_dead)); in RunWithRendezvousArgs()
226 args.rendezvous = rendez_; in RunWithRendezvousArgs()
[all …]
|
| D | executor.h | 164 : rendez_(r), done_cb_(done), pending_(num) {} in ExecutorBarrier()
175 Rendezvous* rendez_ = nullptr;
193 error_rendez = rendez_; in WhenDone()
|
| D | kernel_benchmark_testlib.h | 92 Rendezvous* rendez_ = nullptr; variable
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | single_threaded_executor_test.cc | 53 CHECK(rendez_->Unref()); in ~ExecutorTest()
74 rendez_ = NewLocalRendezvous(); in Create()
94 Rendezvous* rendez_ = nullptr; member in tensorflow::data::__anon06655fe30111::ExecutorTest
|