Searched refs:AddOpsRewriteStage (Results 1 – 3 of 3) sorted by relevance
544 class AddOpsRewriteStage : public ArithmeticNodesGroupOptimizerStage { class546 explicit AddOpsRewriteStage(const GraphOptimizerContext& ctx, in AddOpsRewriteStage() function in tensorflow::grappler::__anon2c8062030111::AddOpsRewriteStage549 ~AddOpsRewriteStage() override = default;4308 pipeline.AddStage<AddOpsRewriteStage>(ctx, ctx_ext); in SimplifyArithmeticOps()
48450 key: "_grappler:ArithmeticOptimizer:AddOpsRewriteStage"