Searched refs:AddOpsRewriteStage (Results 1 – 1 of 1) sorted by relevance
457 class AddOpsRewriteStage : public ArithmeticNodesGroupOptimizerStage { class459 explicit AddOpsRewriteStage(const GraphOptimizerContext& ctx, in AddOpsRewriteStage() function in tensorflow::grappler::__anonbb66c6470111::AddOpsRewriteStage462 ~AddOpsRewriteStage() override = default;3544 pipeline.AddStage<AddOpsRewriteStage>(ctx, ctx_ext); in SimplifyArithmeticOps()