Searched refs:CanContractEdge (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/graphcycles/ |
| D | graphcycles_test.cc | 500 TEST_F(GraphCyclesTest, CanContractEdge) { in TEST_F() argument
507 EXPECT_FALSE(g_.CanContractEdge(1, 3)); in TEST_F()
508 EXPECT_FALSE(g_.CanContractEdge(2, 4)); in TEST_F()
509 EXPECT_TRUE(g_.CanContractEdge(1, 2)); in TEST_F()
510 EXPECT_TRUE(g_.CanContractEdge(2, 3)); in TEST_F()
511 EXPECT_TRUE(g_.CanContractEdge(3, 4)); in TEST_F()
|
| D | graphcycles.h | 91 bool CanContractEdge(int32 a, int32 b);
|
| D | graphcycles.cc | 361 bool GraphCycles::CanContractEdge(int32 a, int32 b) { in CanContractEdge() function in tensorflow::GraphCycles
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_cluster_util_test.cc | 53 EXPECT_FALSE(cycles.CanContractEdge(a.node()->id(), b.node()->id())); in TEST()
72 EXPECT_FALSE(cycles.CanContractEdge(a.node()->id(), b.node()->id())); in TEST()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment.cc | 293 bool CanContractEdge(const SimpleEdge* edge, in CanContractEdge() function
868 if (CanContractEdge(out_edge, graph)) { in SegmentGraph()
|