Searched refs:ContractEdge (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/utils/ |
| D | cycle_detector_test.cc | 69 TEST_F(GraphCyclesTest, ContractEdge) { in TEST_F() argument
77 EXPECT_FALSE(g_.ContractEdge(1, 3).hasValue()); in TEST_F()
81 EXPECT_EQ(*g_.ContractEdge(1, 2), 2); in TEST_F()
87 EXPECT_EQ(*g_.ContractEdge(2, 3), 2); in TEST_F()
|
| D | cycle_detector.cc | 271 llvm::Optional<int32_t> GraphCycles::ContractEdge(int32_t a, int32_t b) { in ContractEdge() function in mlir::GraphCycles
|
| /external/tensorflow/tensorflow/compiler/xla/service/graphcycles/ |
| D | graphcycles_test.cc | 476 TEST_F(GraphCyclesTest, ContractEdge) { in TEST_F() argument
483 EXPECT_FALSE(g_.ContractEdge(1, 3).has_value()); in TEST_F()
488 EXPECT_EQ(g_.ContractEdge(1, 2).value(), 2); in TEST_F()
495 EXPECT_EQ(g_.ContractEdge(2, 3).value(), 2); in TEST_F()
555 node = g.ContractEdge(nodes[i], node).value(); in BM_ContractEdge()
|
| D | graphcycles.h | 88 absl::optional<int32> ContractEdge(int32 a, int32 b);
|
| D | graphcycles.cc | 371 absl::optional<int32> GraphCycles::ContractEdge(int32 a, int32 b) { in ContractEdge() function in tensorflow::GraphCycles
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/utils/ |
| D | cycle_detector.h | 137 llvm::Optional<int32_t> ContractEdge(int32_t a, int32_t b);
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
| D | segment.cc | 574 void ContractEdge(SimpleEdge* edge, SimpleGraph* graph, in ContractEdge() function
898 ContractEdge(e, graph.get(), &remove_edges); in SegmentGraph()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | mhlo_fusion.cc | 360 auto optional_merged_node = cycle_detector_.ContractEdge(from, to); in MergeClusters()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | mark_for_compilation_pass.cc | 397 auto optional_merged_node = cycles_graph_.ContractEdge(from, to); in MergeClusters()
|