Lines Matching refs:DimOutNum
484 int DimOutNum = isl_schedule_node_band_n_member(Node.get()); in createMacroKernel() local
485 std::vector<int> TileSizes(DimOutNum, 1); in createMacroKernel()
486 TileSizes[DimOutNum - 3] = MacroKernelParams.Mc; in createMacroKernel()
487 TileSizes[DimOutNum - 2] = MacroKernelParams.Nc; in createMacroKernel()
488 TileSizes[DimOutNum - 1] = MacroKernelParams.Kc; in createMacroKernel()
491 Node = permuteBandNodeDimensions(Node, DimOutNum - 2, DimOutNum - 1); in createMacroKernel()
492 Node = permuteBandNodeDimensions(Node, DimOutNum - 3, DimOutNum - 1); in createMacroKernel()
958 int DimOutNum = isl_schedule_node_band_n_member(Node.get()); in optimizeMatMulPattern() local
959 assert(DimOutNum > 2 && "In case of the matrix multiplication the loop nest " in optimizeMatMulPattern()
963 Node = permuteBandNodeDimensions(Node, MMI.i, DimOutNum - 3); in optimizeMatMulPattern()
964 int NewJ = MMI.j == DimOutNum - 3 ? MMI.i : MMI.j; in optimizeMatMulPattern()
965 int NewK = MMI.k == DimOutNum - 3 ? MMI.i : MMI.k; in optimizeMatMulPattern()
966 Node = permuteBandNodeDimensions(Node, NewJ, DimOutNum - 2); in optimizeMatMulPattern()
967 NewK = NewK == DimOutNum - 2 ? NewJ : NewK; in optimizeMatMulPattern()
968 Node = permuteBandNodeDimensions(Node, NewK, DimOutNum - 1); in optimizeMatMulPattern()