Searched refs:recv0 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | constant_folding_test.cc | 447 Output recv0 = ops::_Recv(s.WithOpName("recv0"), DT_FLOAT, "recv0", in TEST_F() local
449 auto shape = ops::Shape(s.WithOpName("shape"), recv0); in TEST_F()
452 auto shape_n = ops::ShapeN(s.WithOpName("shape_n"), {recv0, recv1}); in TEST_F()
453 auto rank = ops::Rank(s.WithOpName("rank"), recv0); in TEST_F()
470 Node* recv0 = orig_index.at("recv0"); in TEST_F() local
479 map[recv0->name()].push_back(ps0); in TEST_F()
509 EXPECT_TRUE(e->src() == recv0 || e->src() == recv1 || e->src() == recv2) in TEST_F()
516 EXPECT_TRUE(e->src() == recv0 || e->src() == recv1) << e->src()->name(); in TEST_F()
522 EXPECT_TRUE(e->src() == recv0 || e->src() == recv1) << e->src()->name(); in TEST_F()
530 Output recv0 = ops::_Recv(s.WithOpName("recv0"), DT_FLOAT, "recv0", in TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf-communication.mlir | 229 …%0 = "tf._XlaHostComputeMlir"(%arg0) {recv_key = "recv0", send_key = "send0", tpu_core = 0 : i64} …
|