Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/graphcycles/
Dgraphcycles_test.cc497 TEST_F(GraphCyclesTest, CanContractEdge) { in TEST_F() argument
504 EXPECT_FALSE(g_.CanContractEdge(1, 3)); in TEST_F()
505 EXPECT_FALSE(g_.CanContractEdge(2, 4)); in TEST_F()
506 EXPECT_TRUE(g_.CanContractEdge(1, 2)); in TEST_F()
507 EXPECT_TRUE(g_.CanContractEdge(2, 3)); in TEST_F()
508 EXPECT_TRUE(g_.CanContractEdge(3, 4)); in TEST_F()
Dgraphcycles.h89 bool CanContractEdge(int32 a, int32 b);
Dgraphcycles.cc357 bool GraphCycles::CanContractEdge(int32 a, int32 b) { in CanContractEdge() function in tensorflow::GraphCycles
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc294 bool CanContractEdge(const SimpleEdge* edge, in CanContractEdge() function
505 if (CanContractEdge(out_edge, graph)) { in SegmentGraph()