Searched refs:CanContractEdge (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/jit/graphcycles/ |
D | graphcycles_test.cc | 497 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()
|
D | graphcycles.h | 89 bool CanContractEdge(int32 a, int32 b);
|
D | graphcycles.cc | 357 bool GraphCycles::CanContractEdge(int32 a, int32 b) { in CanContractEdge() function in tensorflow::GraphCycles
|
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
D | segment.cc | 294 bool CanContractEdge(const SimpleEdge* edge, in CanContractEdge() function 505 if (CanContractEdge(out_edge, graph)) { in SegmentGraph()
|