Searched refs:indexAttr (Results 1 – 3 of 3) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/SPIRV/ |
| D | SPIRVOps.cpp | 617 for (auto indexAttr : indicesArrayAttr) { in getElementType() local
618 auto indexIntAttr = indexAttr.dyn_cast<IntegerAttr>(); in getElementType()
621 << indexAttr << "'"; in getElementType()
1354 auto indexAttr = builder.getI32ArrayAttr(indices); in build() local
1356 getElementType(composite.getType(), indexAttr, state.location); in build()
1360 build(builder, state, elementType, composite, indexAttr); in build()
1416 auto indexAttr = builder.getI32ArrayAttr(indices); in build() local
1417 build(builder, state, composite.getType(), object, composite, indexAttr); in build()
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorOps.cpp | 2901 IntegerAttr indexAttr; in parseTupleGetOp() local
2906 parser.parseAttribute(indexAttr, indexType, indexAttrName, in parseTupleGetOp()
2912 if (indexAttr.getInt() < 0 || in parseTupleGetOp()
2913 indexAttr.getInt() >= static_cast<int64_t>(tupleType.size())) in parseTupleGetOp()
2915 parser.addTypeToList(tupleType.getType(indexAttr.getInt()), result.types); in parseTupleGetOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 646 if (!first_element_op || first_element_op.indexAttr().getInt() != 0) in matchAndRewrite()
656 element_op.indexAttr().getInt() != element_and_idx.index() + 1 || in matchAndRewrite()
|