Searched refs:DimOutNum (Results 1 – 1 of 1) sorted by relevance
484 int DimOutNum = isl_schedule_node_band_n_member(Node.get()); in createMacroKernel() local485 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() local964 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()[all …]