Searched refs:FuseSquaredDiffStage (Results 1 – 1 of 1) sorted by relevance
1781 class FuseSquaredDiffStage : public ArithmeticOptimizerStage { class1783 explicit FuseSquaredDiffStage(const GraphOptimizerContext& ctx, in FuseSquaredDiffStage() function in tensorflow::grappler::__anonbb66c6470111::FuseSquaredDiffStage1786 ~FuseSquaredDiffStage() override = default;3595 pipeline.AddStage<FuseSquaredDiffStage>(ctx, ctx_ext); in SimplifyArithmeticOps()