Searched refs:FuseSquaredDiffStage (Results 1 – 1 of 1) sorted by relevance
1869 class FuseSquaredDiffStage : public ArithmeticOptimizerStage { class1871 explicit FuseSquaredDiffStage(const GraphOptimizerContext& ctx, in FuseSquaredDiffStage() function in tensorflow::grappler::__anon327bfa1e0111::FuseSquaredDiffStage1874 ~FuseSquaredDiffStage() override = default;3737 pipeline.AddStage<FuseSquaredDiffStage>(ctx, ctx_ext); in SimplifyArithmeticOps()