Home
last modified time | relevance | path

Searched full:type_var (Results 1 – 9 of 9) sorted by relevance

/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/
H A DLinalgNamedStructuredOps.yaml19 type_var: T1
24 type_var: U
42 type_var: U
60 type_var: T1
65 type_var: U
92 type_var: U
109 type_var: T1
114 type_var: T1
144 type_var: T1
149 type_var
[all...]
/llvm-project/mlir/python/mlir/dialects/linalg/opdsl/lang/
H A Dscalar_expr.py40 type_var: Optional["TypeVar"],
48 self.type_var = type_var
57 f"ScalarFn<{self.kind.name}.{name}>(type_var={self.type_var}, "
139 if self.scalar_fn.type_var:
140 scalar_fn_dict["type_var"] = self.scalar_fn.type_var.name
H A Dcomprehension.py147 type_var: Optional[TypeVar],
155 self.type_var = type_var
163 return ScalarFn(self.kind, self.name, attr_name, self.type_var, args).expr()
173 f"{self.kind.name}.{name}(type_var={self.type_var}, "
381 def __call__(self, type_var: TypeVar, arg: TensorExpression) -> "TensorFn":
382 return TensorFn(FunctionKind.TYPE, self.fn_name, None, type_var, [arg])
483 type_var: Optional[TypeVar] = None,
489 if type_var an
[all...]
H A Dconfig.py66 def type_var(self) -> TypeVar: member in OperandDefConfig
67 return self.operand_def.type_var
71 if self.type_var:
72 self_dict["type_var"] = self.type_var.name
H A Demitter.py381 operand_values = [expr.scalar_fn.type_var.name] + operand_values
603 name = operand_config.type_var.name
/llvm-project/mlir/test/python/dialects/linalg/opdsl/
H A Darguments.py11 # CHECK: type_var: T
15 # CHECK: type_var: T
19 # CHECK: type_var: U
48 # CHECK: type_var: T
59 # CHECK: type_var: T
63 # CHECK: type_var: T
H A Dassignments.py23 # CHECK: type_var: U
29 # CHECK: type_var: U
62 # CHECK: type_var: T
68 # CHECK: type_var: T
74 # CHECK: type_var: T
/llvm-project/mlir/test/mlir-linalg-ods-gen/
H A Dtest-linalg-ods-yaml-gen.yaml26 type_var: T
50 type_var: T
58 type_var: T
129 type_var: T
134 type_var: T
216 type_var: T1
220 type_var: U
234 type_var: U
270 type_var: T
353 type_var: T1
[all …]
/llvm-project/mlir/tools/mlir-linalg-ods-gen/
H A Dmlir-linalg-ods-yaml-gen.cpp178 /// - `type_var`: The symbolic type variable that binds to the element or self
197 io.mapOptional("type_var", info.typeVar); in mapping()
302 io.mapOptional("type_var", info.typeVar); in mapping()