Home
last modified time | relevance | path

Searched refs:ContractEdge (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dxla_cluster_util_test.cc49 EXPECT_FALSE(cycles.ContractEdge(a.node()->id(), b.node()->id())); in TEST()
68 EXPECT_FALSE(cycles.ContractEdge(a.node()->id(), b.node()->id())); in TEST()
Dxla_fusion_optimizer.cc292 if (!cycles.ContractEdge(from, to)) continue; in Optimize()
Dmark_for_compilation_pass.cc1247 if (!cycles.ContractEdge(from, to)) continue; in RunImpl()
/external/tensorflow/tensorflow/compiler/jit/graphcycles/
Dgraphcycles_test.cc475 TEST_F(GraphCyclesTest, ContractEdge) { in TEST_F() argument
482 EXPECT_FALSE(g_.ContractEdge(1, 3)); in TEST_F()
486 EXPECT_TRUE(g_.ContractEdge(1, 2)); in TEST_F()
492 EXPECT_TRUE(g_.ContractEdge(1, 3)); in TEST_F()
Dgraphcycles.h86 bool ContractEdge(int32 a, int32 b);
Dgraphcycles.cc367 bool GraphCycles::ContractEdge(int32 a, int32 b) { in ContractEdge() function in tensorflow::GraphCycles
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc346 void ContractEdge(SimpleEdge* edge, SimpleGraph* graph, in ContractEdge() function
531 ContractEdge(e, graph.get(), &remove_edges); in SegmentGraph()