Searched refs:FoldTransposeIntoMatMul (Results 1 – 2 of 2) sorted by relevance
2276 class FoldTransposeIntoMatMul : public ArithmeticOptimizerStage { class2278 explicit FoldTransposeIntoMatMul(const GraphOptimizerContext& ctx, in FoldTransposeIntoMatMul() function in tensorflow::grappler::__anon73f394250111::FoldTransposeIntoMatMul2281 ~FoldTransposeIntoMatMul() override = default;3688 pipeline.AddStage<FoldTransposeIntoMatMul>(ctx, ctx_ext); in SimplifyArithmeticOps()
715 TEST_F(ArithmeticOptimizerTest, FoldTransposeIntoMatMul) { in TEST_F() argument