Searched refs:received_keys (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | rendezvous_util_test.cc | 60 std::vector<Tensor> received_keys; in TEST_F() local
63 &received_keys, [&n](const Status& status) { n.Notify(); }); in TEST_F()
66 EXPECT_EQ(2, received_keys.size()); in TEST_F()
67 EXPECT_EQ("hello1", V(received_keys[0])); in TEST_F()
68 EXPECT_EQ("hello2", V(received_keys[1])); in TEST_F()
74 std::vector<Tensor> received_keys; in TEST_F() local
77 &received_keys, [&n](const Status& status) { n.Notify(); }); in TEST_F()
85 EXPECT_EQ(2, received_keys.size()); in TEST_F()
86 EXPECT_EQ("hello1", V(received_keys[0])); in TEST_F()
87 EXPECT_EQ("hello2", V(received_keys[1])); in TEST_F()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 372 std::vector<Tensor>* received_keys = new std::vector<Tensor>; in RecvOutputsAsync() local
374 received_keys->reserve(out->size()); in RecvOutputsAsync()
377 received_keys->push_back(p.second); in RecvOutputsAsync()
380 rendezvous, nullptr, {}, keys, received_keys, in RecvOutputsAsync()
381 [done, rendezvous, received_keys, out, keys](const Status s) { in RecvOutputsAsync()
384 (*out)[keys[i]] = (*received_keys)[i]; in RecvOutputsAsync()
386 delete received_keys; in RecvOutputsAsync()
|