| /external/tensorflow/tensorflow/core/framework/ |
| D | rendezvous_test.cc | 69 rendez_ = NewLocalRendezvous(); in LocalRendezvousTest()
72 ~LocalRendezvousTest() override { rendez_->Unref(); } in ~LocalRendezvousTest()
78 Rendezvous* rendez_; member in tensorflow::__anon2f0f779b0111::LocalRendezvousTest
119 TF_ASSERT_OK(rendez_->Send(KeyFoo(), args, V("hello"), false)); in TEST_F()
122 TF_ASSERT_OK(rendez_->Recv(KeyFoo(), args, &val, &is_dead)); in TEST_F()
130 TF_ASSERT_OK(rendez_->Send(KeyFoo(), args, V("hello"), false)); in TEST_F()
135 TF_ASSERT_OK(rendez_->Recv(KeyFoo(), args, &val, &is_dead)); in TEST_F()
144 TF_ASSERT_OK(rendez_->Recv(KeyFoo(), args, &t, &is_dead)); in TEST_F()
145 TF_ASSERT_OK(rendez_->Send(KeyBar(), args, t, is_dead)); in TEST_F()
151 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 | 54 CHECK(rendez_->Unref()); in ~ExecutorTest()
74 rendez_ = NewLocalRendezvous(); in Create()
91 Rendezvous* rendez_ = nullptr; member in tensorflow::ExecutorTest
154 TF_ASSERT_OK(rendez_->Send(Key(ALICE, kIncarnation, BOB, "a"), args, V(1.0), in TEST_F()
156 TF_ASSERT_OK(rendez_->Send(Key(ALICE, kIncarnation, BOB, "b"), args, V(1.0), in TEST_F()
158 TF_ASSERT_OK(Run(rendez_)); in TEST_F()
162 rendez_->Recv(Key(BOB, kIncarnation, ALICE, "c"), args, &out, &is_dead)); in TEST_F()
187 rendez_->Send(Key(ALICE, kIncarnation, BOB, "a"), args, V(1.0), false)); in TEST_F()
188 TF_ASSERT_OK(Run(rendez_)); in TEST_F()
192 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::__anonf8c91bd60111::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 | 70 rendez_ = NewLocalRendezvous(); in Benchmark()
72 rendez_ = rendez; in Benchmark()
94 args.rendezvous = rendez_; in Benchmark()
105 rendez_->Unref(); in ~Benchmark()
153 args.rendezvous = rendez_; in RunWithArgs()
162 TF_CHECK_OK(rendez_->Send(parsed, Rendezvous::Args(), p.second, false)); in RunWithArgs()
168 TF_CHECK_OK(rendez_->Recv(parsed, Rendezvous::Args(), &unused, &is_dead)); in RunWithArgs()
179 TF_CHECK_OK(rendez_->Send(parsed, Rendezvous::Args(), p.second, false)); in RunWithArgs()
185 TF_CHECK_OK(rendez_->Recv(parsed, Rendezvous::Args(), &unused, &is_dead)); in RunWithArgs()
|
| D | executor.h | 161 : rendez_(r), done_cb_(done), pending_(num) {} in ExecutorBarrier()
172 Rendezvous* rendez_ = nullptr;
190 error_rendez = rendez_; in WhenDone()
|
| D | kernel_benchmark_testlib.h | 59 Rendezvous* rendez_ = nullptr; variable
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | single_threaded_executor_test.cc | 52 CHECK(rendez_->Unref()); in ~ExecutorTest()
72 rendez_ = NewLocalRendezvous(); in Create()
92 Rendezvous* rendez_ = nullptr; member in tensorflow::data::__anonf76da2850111::ExecutorTest
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 140 rendez_(rendez), in KernelAndDeviceOp()
171 Rendezvous* const rendez_; variable
|
| D | kernel_and_device.cc | 241 params.rendezvous = rendez_; in Run()
|