Home
last modified time | relevance | path

Searched refs:IsMultiInput (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/graph_analyzer/
Dgen_node_test.cc121 EXPECT_THAT(gn3->IsMultiInput(GenNode::Port(true, 0)), Eq(false)); in TEST()
124 EXPECT_THAT(gn3->IsMultiInput(GenNode::Port(true, -1)), Eq(true)); in TEST()
163 EXPECT_THAT(gn3->IsMultiInput(GenNode::Port(true, 0)), Eq(true)); in TEST()
197 EXPECT_THAT(gn2->IsMultiInput(GenNode::Port(false, 0)), Eq(false)); in TEST()
199 EXPECT_THAT(gn3->IsMultiInput(GenNode::Port(true, 0)), Eq(true)); in TEST()
234 EXPECT_THAT(gn3->IsMultiInput(GenNode::Port(true, 0)), Eq(false)); in TEST()
268 EXPECT_THAT(gn3->IsMultiInput(GenNode::Port(true, 0)), Eq(false)); in TEST()
319 EXPECT_THAT(gn5->IsMultiInput(GenNode::Port(true, 1)), Eq(false)); in TEST()
320 EXPECT_THAT(gn5->IsMultiInput(GenNode::Port(true, 2)), Eq(false)); in TEST()
Dgraph_analyzer.cc117 } else if (node->IsMultiInput(port)) { in ExtendSubgraph()
120 } else if (neighbor.node->IsMultiInput(neighbor.port)) { in ExtendSubgraph()
261 if (sit.GetNode()->IsMultiInput(sit.GetPort())) { in HasInvalidMultiInputs()
Dgen_node.h143 bool IsMultiInput(Port port) const;
Dgen_node.cc125 bool GenNode::IsMultiInput(Port port) const { in IsMultiInput() function in tensorflow::grappler::graph_analyzer::GenNode