Searched refs:send_from_host (Results 1 – 2 of 2) sorted by relevance
88 Node *send_from_host = node_name_image["outside_compilation_cluster__0_send"]; in TEST() local89 EXPECT_NE(send_from_host, nullptr); in TEST()92 GetNodeAttr(send_from_host->attrs(), "Tinputs", &send_from_host_dtypes)); in TEST()111 if (e->IsControlEdge() && e->dst() == send_from_host) { in TEST()144 Node *send_from_host = node_name_image["outside_compilation_cluster__0_send"]; in TEST() local145 EXPECT_EQ(send_from_host, nullptr); in TEST()170 Node *send_from_host = node_name_image["outside_compilation_cluster__0_send"]; in TEST() local171 EXPECT_NE(send_from_host, nullptr); in TEST()194 Node *send_from_host = node_name_image["outside_compilation_cluster__0_send"]; in TEST() local195 EXPECT_EQ(send_from_host, nullptr); in TEST()
1219 Node* send_from_host = nullptr; in RewriteShapeInferenceGraph() local1222 send_from_host = n; in RewriteShapeInferenceGraph()1226 if (!send_from_host) { in RewriteShapeInferenceGraph()1235 if (n->name() == send_from_host->name()) { in RewriteShapeInferenceGraph()1291 send_from_host = node_map[send_node_in_host_graph]; in RewriteShapeInferenceGraph()1307 std::unordered_set<const Node*>{send_from_host}); in RewriteShapeInferenceGraph()