Lines Matching refs:attr

33 bool mlirAttributeIsASparseTensorEncodingAttr(MlirAttribute attr) {  in mlirAttributeIsASparseTensorEncodingAttr()  argument
34 return unwrap(attr).isa<SparseTensorEncodingAttr>(); in mlirAttributeIsASparseTensorEncodingAttr()
51 MlirAffineMap mlirSparseTensorEncodingAttrGetDimOrdering(MlirAttribute attr) { in mlirSparseTensorEncodingAttrGetDimOrdering() argument
52 return wrap(unwrap(attr).cast<SparseTensorEncodingAttr>().getDimOrdering()); in mlirSparseTensorEncodingAttrGetDimOrdering()
55 intptr_t mlirSparseTensorEncodingGetNumDimLevelTypes(MlirAttribute attr) { in mlirSparseTensorEncodingGetNumDimLevelTypes() argument
56 return unwrap(attr).cast<SparseTensorEncodingAttr>().getDimLevelType().size(); in mlirSparseTensorEncodingGetNumDimLevelTypes()
60 mlirSparseTensorEncodingAttrGetDimLevelType(MlirAttribute attr, intptr_t pos) { in mlirSparseTensorEncodingAttrGetDimLevelType() argument
62 unwrap(attr).cast<SparseTensorEncodingAttr>().getDimLevelType()[pos]); in mlirSparseTensorEncodingAttrGetDimLevelType()
65 int mlirSparseTensorEncodingAttrGetPointerBitWidth(MlirAttribute attr) { in mlirSparseTensorEncodingAttrGetPointerBitWidth() argument
66 return unwrap(attr).cast<SparseTensorEncodingAttr>().getPointerBitWidth(); in mlirSparseTensorEncodingAttrGetPointerBitWidth()
69 int mlirSparseTensorEncodingAttrGetIndexBitWidth(MlirAttribute attr) { in mlirSparseTensorEncodingAttrGetIndexBitWidth() argument
70 return unwrap(attr).cast<SparseTensorEncodingAttr>().getIndexBitWidth(); in mlirSparseTensorEncodingAttrGetIndexBitWidth()