Searched refs:ReplaceMulWithSquare (Results 1 – 2 of 2) sorted by relevance
2312 class ReplaceMulWithSquare : public ArithmeticOptimizerStage { class2314 explicit ReplaceMulWithSquare(const GraphOptimizerContext& ctx, in ReplaceMulWithSquare() function in tensorflow::grappler::__anonbb66c6470111::ReplaceMulWithSquare2317 ~ReplaceMulWithSquare() override = default;3568 pipeline.AddStage<ReplaceMulWithSquare>(ctx, ctx_ext); in SimplifyArithmeticOps()
213 TEST_F(ArithmeticOptimizerTest, ReplaceMulWithSquare) { in TEST_F() argument