Searched refs:AddOpsRewriteStage (Results 1 – 2 of 2) sorted by relevance
482 class AddOpsRewriteStage : public ArithmeticNodesGroupOptimizerStage { class484 explicit AddOpsRewriteStage(const GraphOptimizerContext& ctx, in AddOpsRewriteStage() function in tensorflow::grappler::__anon327bfa1e0111::AddOpsRewriteStage487 ~AddOpsRewriteStage() override = default;3682 pipeline.AddStage<AddOpsRewriteStage>(ctx, ctx_ext); in SimplifyArithmeticOps()
48450 key: "_grappler:ArithmeticOptimizer:AddOpsRewriteStage"