Searched refs:recv2 (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_subgraphs_pass_test.cc | 1061 Node* recv2 = RecvAtHost( in TEST() local
1064 Node* g = Binary(e, ops::NodeOut(recv2, 0), in TEST()
1069 Node* h = Binary(ops::NodeOut(recv2, 1), e, in TEST()
1149 Node* recv2 = RecvAtHost( in TEST() local
1152 Node* g = Binary(e, ops::NodeOut(recv2, 0), in TEST()
1155 .WithControlInputs({recv2, e}) in TEST()
1158 Node* h = Binary(ops::NodeOut(recv2, 1), e, in TEST()
1169 .WithControlInputs({recv1, send1, recv2, send2}), in TEST()
1318 Node* recv2 = RecvAtHost( in TEST() local
1321 Node* h = Binary(recv2, ops::NodeOut(recv2, 1), in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding_test.cc | 414 auto recv2 = ops::_Recv(s.WithOpName("recv2"), DT_FLOAT, "recv2", "sender", in TEST_F() local
416 auto c2 = ops::Const<int>(s.WithControlDependencies(recv2), 3); in TEST_F()
429 Node* recv2 = index.at("recv2"); in TEST_F() local
439 EXPECT_TRUE(e->src() == recv1 || e->src() == recv2) << e->src()->name(); in TEST_F()
455 auto recv2 = ops::_Recv(s.WithOpName("recv2"), DT_FLOAT, "recv2", "sender", in TEST_F() local
457 auto c = ops::Const<int>(s.WithControlDependencies(recv2), 3); in TEST_F()
489 Node* recv2 = index.at("recv2"); in TEST_F() local
509 EXPECT_TRUE(e->src() == recv0 || e->src() == recv1 || e->src() == recv2) in TEST_F()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition_test.cc | 366 auto recv2 = in TEST_F() local
369 Combine(scope_b_.WithOpName("B2"), recv2, b1); in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 2118 NodeDef recv2; in TEST_F() local
2122 NodeSetUp("Recv2", kRecv, kCPU1, 2004, &recv2); in TEST_F()
2137 manager.AddNode(&recv2); in TEST_F()
|