Searched refs:FoldTransposeIntoMatMul (Results 1 – 2 of 2) sorted by relevance
2165 class FoldTransposeIntoMatMul : public ArithmeticOptimizerStage { class2167 explicit FoldTransposeIntoMatMul(const GraphOptimizerContext& ctx, in FoldTransposeIntoMatMul() function in tensorflow::grappler::__anonbb66c6470111::FoldTransposeIntoMatMul2170 ~FoldTransposeIntoMatMul() override = default;3550 pipeline.AddStage<FoldTransposeIntoMatMul>(ctx, ctx_ext); in SimplifyArithmeticOps()
790 TEST_F(ArithmeticOptimizerTest, FoldTransposeIntoMatMul) { in TEST_F() argument