Searched defs:EliminateExtraPermuteOps (Results 1 – 2 of 2) sorted by relevance
1560 TEST(EliminateExtraPermuteOps, FusesSumCorrectly) { in TEST() argument1583 TEST(EliminateExtraPermuteOps, DoesNotFuseSumWrongDim) { in TEST() argument1601 TEST(EliminateExtraPermuteOps, DoesNotFuseSumNonConstantDim) { in TEST() argument1619 TEST(EliminateExtraPermuteOps, FusesSoftmaxCorrectly) { in TEST() argument1645 TEST(EliminateExtraPermuteOps, DoesNotFuseSoftmaxWrongPermuteDim) { in TEST() argument1661 TEST(EliminateExtraPermuteOps, DoesNotFuseSoftmaxWrongSoftmaxDim) { in TEST() argument
1220 void EliminateExtraPermuteOps(std::shared_ptr<Graph>& graph) { in EliminateExtraPermuteOps() function