Searched refs:ReplaceMulWithSquare (Results 1 – 2 of 2) sorted by relevance
2425 class ReplaceMulWithSquare : public ArithmeticOptimizerStage { class2427 explicit ReplaceMulWithSquare(const GraphOptimizerContext& ctx, in ReplaceMulWithSquare() function in tensorflow::grappler::__anon73f394250111::ReplaceMulWithSquare2430 ~ReplaceMulWithSquare() override = default;3706 pipeline.AddStage<ReplaceMulWithSquare>(ctx, ctx_ext); in SimplifyArithmeticOps()
109 TEST_F(ArithmeticOptimizerTest, ReplaceMulWithSquare) { in TEST_F() argument