Searched refs:NewK (Results 1 – 2 of 2) sorted by relevance
965 int NewK = MMI.k == DimOutNum - 3 ? MMI.i : MMI.k; in optimizeMatMulPattern() local967 NewK = NewK == DimOutNum - 2 ? NewJ : NewK; in optimizeMatMulPattern()968 Node = permuteBandNodeDimensions(Node, NewK, DimOutNum - 1); in optimizeMatMulPattern()
128 ArgEffect withKind(ArgEffectKind NewK) { in withKind() argument129 return ArgEffect(NewK, O); in withKind()