Searched refs:GraphPruningPass (Results 1 – 1 of 1) sorted by relevance
38 class GraphPruningPass class39 : public TF::ExecutorGraphPruningPassBase<GraphPruningPass> {41 GraphPruningPass() = default;42 explicit GraphPruningPass(llvm::ArrayRef<std::string> ops_to_preserve);108 GraphPruningPass::GraphPruningPass( in GraphPruningPass() function in mlir::tf_executor::__anond7bd4ed10111::GraphPruningPass113 void GraphPruningPass::runOnFunction() { in runOnFunction()123 bool GraphPruningPass::ShouldPreserveOp(Operation* op) { in ShouldPreserveOp()128 bool GraphPruningPass::ShouldPreserveIsland(IslandOp island) { in ShouldPreserveIsland()137 void GraphPruningPass::PruneGraph(GraphOp graph) { in PruneGraph()178 return std::make_unique<GraphPruningPass>(ops_to_preserve); in CreateTFExecutorGraphPruningPass()