262 KernelDim3 LaunchOp::getBlockIds() {  in getBlockIds()
265   return KernelDim3{args[0], args[1], args[2]};  in getBlockIds()
268 KernelDim3 LaunchOp::getThreadIds() {  in getThreadIds()
271   return KernelDim3{args[3], args[4], args[5]};  in getThreadIds()
274 KernelDim3 LaunchOp::getGridSize() {  in getGridSize()
277   return KernelDim3{args[6], args[7], args[8]};  in getGridSize()
280 KernelDim3 LaunchOp::getBlockSize() {  in getBlockSize()
283   return KernelDim3{args[9], args[10], args[11]};  in getBlockSize()
286 KernelDim3 LaunchOp::getGridSizeOperandValues() {  in getGridSizeOperandValues()
287   return KernelDim3{getOperand(0), getOperand(1), getOperand(2)};  in getGridSizeOperandValues()
290 KernelDim3 LaunchOp::getBlockSizeOperandValues() {  in getBlockSizeOperandValues()
291   return KernelDim3{getOperand(3), getOperand(4), getOperand(5)};  in getBlockSizeOperandValues()
328 static void printSizeAssignment(OpAsmPrinter &p, KernelDim3 size,  in printSizeAssignment()
329                                 KernelDim3 operands, KernelDim3 ids) {  in printSizeAssignment()
431                          GPUFuncOp kernelFunc, KernelDim3 gridSize,  in build()
432                          KernelDim3 blockSize, ValueRange kernelOperands) {  in build()
462 KernelDim3 LaunchFuncOp::getGridSizeOperandValues() {  in getGridSizeOperandValues()
464   return KernelDim3{operands[0], operands[1], operands[2]};  in getGridSizeOperandValues()
467 KernelDim3 LaunchFuncOp::getBlockSizeOperandValues() {  in getBlockSizeOperandValues()
469   return KernelDim3{operands[3], operands[4], operands[5]};  in getBlockSizeOperandValues()