Searched refs:dimAttr (Results 1 – 5 of 5) sorted by relevance
103 auto dimAttr = op->getAttrOfType<StringAttr>("dimension"); in getLaunchConfigIndex() local104 if (!dimAttr) { in getLaunchConfigIndex()107 if (dimAttr.getValue() == "x") { in getLaunchConfigIndex()109 } else if (dimAttr.getValue() == "y") { in getLaunchConfigIndex()111 } else if (dimAttr.getValue() == "z") { in getLaunchConfigIndex()
47 for (auto dimAttr : mapping) { in setMappingAttr() local48 gpu::Processor processor = getProcessor(dimAttr); in setMappingAttr()
658 auto dimAttr = builder.getIndexAttr(dim); in build() local660 Value dim = builder.create<ConstSizeOp>(loc, dimAttr); in build()664 builder.create<ConstantOp>(loc, builder.getIndexType(), dimAttr); in build()
472 auto dimAttr = sparseAttr[t].dyn_cast_or_null<ArrayAttr>(); in verify() local473 if (!dimAttr) in verify()477 if (dimAttr.size() != rank) in verify()482 if (isDenseDim(dimAttr[d])) { in verify()484 } else if (isSparseDim(dimAttr[d])) { in verify()
286 auto dimAttr = sparseAttr[t].cast<ArrayAttr>(); in findSparseAnnotations() local291 if (isSparseDim(dimAttr[d])) { in findSparseAnnotations()295 assert(isDenseDim(dimAttr[d])); in findSparseAnnotations()