Home
last modified time | relevance | path

Searched refs:HasFanin (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dgraph_view_test.cc245 TEST_F(GraphViewTest, HasFanin) { in TEST_F() argument
259 EXPECT_EQ(graph.HasFanin(*d_node, {"a", Graph::kControlSlot}), true); in TEST_F()
260 EXPECT_EQ(graph.HasFanin(*d_node, {"a", 0}), false); in TEST_F()
261 EXPECT_EQ(graph.HasFanin(*d_node, {"b", 0}), true); in TEST_F()
262 EXPECT_EQ(graph.HasFanin(*d_node, {"b", Graph::kControlSlot}), false); in TEST_F()
263 EXPECT_EQ(graph.HasFanin(*d_node, {"c", 0}), true); in TEST_F()
264 EXPECT_EQ(graph.HasFanin(*d_node, {"c", Graph::kControlSlot}), false); in TEST_F()
Dgraph_view.h182 bool HasFanin(const NodeDefT& node, const TensorId& fanin) const { in HasFanin() function
/external/tensorflow/tensorflow/core/grappler/utils/
Dgraph_view.h95 bool HasFanin(const FanoutView& fanin) const override;
224 bool HasFanin(const MutableFanoutView& fanin) const override;
Dgraph_view_test.cc657 TYPED_TEST(TypedNodeViewTest, HasFanin) { in TYPED_TEST() argument
672 EXPECT_TRUE(a_node->HasFanin({&graph_view, b_node->node_index(), 2})); in TYPED_TEST()
674 EXPECT_FALSE(a_node->HasFanin({&graph_view, c_node->node_index(), 4})); in TYPED_TEST()
676 EXPECT_TRUE(a_node->HasFanin( in TYPED_TEST()
679 EXPECT_FALSE(a_node->HasFanin( in TYPED_TEST()
682 EXPECT_FALSE(a_node->HasFanin({&graph_view, a_node->node_index(), 0})); in TYPED_TEST()
683 EXPECT_FALSE(a_node->HasFanin( in TYPED_TEST()
829 node_view->HasFanin({actual_fanin_node, expected_fanin.index()})); in CompareGraphViewWithGraph()
860 EXPECT_TRUE(node_view->HasFanin(actual_fanin)); in CompareGraphViewWithGraph()
2713 node->HasFanin({&graph_view, fanin->node_index(), 0}); in BM_GraphViewTHasRegularFanin()
[all …]
Dgraph_view_internal.h220 virtual bool HasFanin(const FanoutViewT& fanin) const = 0;
Dgraph_view.cc48 bool NodeView::HasFanin(const FanoutView& fanin) const { in HasFanin() function in tensorflow::grappler::utils::NodeView
183 bool MutableNodeView::HasFanin(const MutableFanoutView& fanin) const { in HasFanin() function in tensorflow::grappler::utils::MutableNodeView