}];
let arguments = (ins AnyTypeOf<[AnyShaped, Shape_ValueShapeType]>:$operand,
- Shape_ShapeType:$shape);
+ Shape_ShapeOrExtentTensorType:$shape);
let results = (outs Shape_ValueShapeType:$result);
let assemblyFormat = "operands attr-dict `:` type($operand) `,` type($shape)";
OptionalAttr<StrAttr>:$sym_visibility);
let regions = (region AnyRegion:$body);
+ let builders = [OpBuilder<(ins
+ "StringRef":$name, "FunctionType":$type,
+ CArg<"ArrayRef<NamedAttribute>", "{}">:$attrs,
+ CArg<"ArrayRef<DictionaryAttr>", "{}">:$argAttrs)
+ >];
+
let extraClassDeclaration = [{
+ static FuncOp create(Location location, StringRef name, FunctionType type,
+ ArrayRef<NamedAttribute> attrs = {});
+ static FuncOp create(Location location, StringRef name, FunctionType type,
+ Operation::dialect_attr_range attrs);
+ static FuncOp create(Location location, StringRef name, FunctionType type,
+ ArrayRef<NamedAttribute> attrs,
+ ArrayRef<DictionaryAttr> argAttrs);
//===------------------------------------------------------------------===//
// CallableOpInterface
//===------------------------------------------------------------------===//
// FuncOp
//===----------------------------------------------------------------------===//
+FuncOp FuncOp::create(Location location, StringRef name, FunctionType type,
+ ArrayRef<NamedAttribute> attrs) {
+ OpBuilder builder(location->getContext());
+ OperationState state(location, getOperationName());
+ FuncOp::build(builder, state, name, type, attrs);
+ return cast<FuncOp>(Operation::create(state));
+}
+FuncOp FuncOp::create(Location location, StringRef name, FunctionType type,
+ Operation::dialect_attr_range attrs) {
+ SmallVector<NamedAttribute, 8> attrRef(attrs);
+ return create(location, name, type, llvm::makeArrayRef(attrRef));
+}
+FuncOp FuncOp::create(Location location, StringRef name, FunctionType type,
+ ArrayRef<NamedAttribute> attrs,
+ ArrayRef<DictionaryAttr> argAttrs) {
+ FuncOp func = create(location, name, type, attrs);
+ func.setAllArgAttrs(argAttrs);
+ return func;
+}
+
+void FuncOp::build(OpBuilder &builder, OperationState &state, StringRef name,
+ FunctionType type, ArrayRef<NamedAttribute> attrs,
+ ArrayRef<DictionaryAttr> argAttrs) {
+ state.addAttribute(FuncOp::getSymNameAttrName(state.name),
+ builder.getStringAttr(name));
+ state.addAttribute(FuncOp::getFunctionTypeAttrName(state.name),
+ TypeAttr::get(type));
+ state.attributes.append(attrs.begin(), attrs.end());
+ state.addRegion();
+
+ if (argAttrs.empty())
+ return;
+ assert(type.getNumInputs() == argAttrs.size());
+ function_interface_impl::addArgAndResultAttrs(builder, state, argAttrs,
+ /*resultAttrs=*/llvm::None);
+}
+
ParseResult FuncOp::parse(OpAsmParser &parser, OperationState &result) {
auto buildFuncType =
[](Builder &builder, ArrayRef<Type> argTypes, ArrayRef<Type> results,
return %2 : !shape.shape
}
+func.func @shape_with_shape_extent_tensor_type(%a : tensor<?x?x?xf32>, %b : !shape.value_shape) -> !shape.value_shape {
+ %0 = shape.shape_of %a : tensor<?x?x?xf32> -> tensor<3xindex>
+ %1 = shape.with_shape %b, %0 : !shape.value_shape, tensor<3xindex>
+ return %1 : !shape.value_shape
+}
+
func.func @any_on_shape(%a : !shape.shape, %b : !shape.shape, %c : !shape.shape)
-> !shape.shape {
%result = shape.any %a, %b, %c