- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for ShouldPreserveIsland (0.16 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/graph_pruning.cc
public: GraphPruningPass() = default; explicit GraphPruningPass(llvm::ArrayRef<std::string> ops_to_preserve); void runOnOperation() override; private: bool ShouldPreserveOp(Operation* op); bool ShouldPreserveIsland(IslandOp island); void PruneGraph(GraphOp graph); llvm::SmallDenseSet<mlir::StringAttr, 4> ops_to_preserve_ids_; }; // Checks if a tf_executor.Graph can be pruned.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 7.6K bytes - Viewed (0)