Lines Matching refs:DimOutNum
840 int DimOutNum = isl_schedule_node_band_n_member(Node.get()); in createMacroKernel() local
841 std::vector<int> TileSizes(DimOutNum, 1); in createMacroKernel()
842 TileSizes[DimOutNum - 3] = MacroKernelParams.Mc; in createMacroKernel()
843 TileSizes[DimOutNum - 2] = MacroKernelParams.Nc; in createMacroKernel()
844 TileSizes[DimOutNum - 1] = MacroKernelParams.Kc; in createMacroKernel()
847 Node = permuteBandNodeDimensions(Node, DimOutNum - 2, DimOutNum - 1); in createMacroKernel()
848 Node = permuteBandNodeDimensions(Node, DimOutNum - 3, DimOutNum - 1); in createMacroKernel()
1282 int DimOutNum = isl_schedule_node_band_n_member(Node.get()); in optimizeMatMulPattern() local
1283 assert(DimOutNum > 2 && "In case of the matrix multiplication the loop nest " in optimizeMatMulPattern()
1287 Node = permuteBandNodeDimensions(Node, MMI.i, DimOutNum - 3); in optimizeMatMulPattern()
1288 int NewJ = MMI.j == DimOutNum - 3 ? MMI.i : MMI.j; in optimizeMatMulPattern()
1289 int NewK = MMI.k == DimOutNum - 3 ? MMI.i : MMI.k; in optimizeMatMulPattern()
1290 Node = permuteBandNodeDimensions(Node, NewJ, DimOutNum - 2); in optimizeMatMulPattern()
1291 NewK = NewK == DimOutNum - 2 ? NewJ : NewK; in optimizeMatMulPattern()
1292 Node = permuteBandNodeDimensions(Node, NewK, DimOutNum - 1); in optimizeMatMulPattern()