Home
last modified time | relevance | path

Searched refs:type_var (Results 1 – 6 of 6) sorted by relevance

/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/
H A DLinalgNamedStructuredOps.yaml17 type_var: T1
22 type_var: U
40 type_var: U
58 type_var: T1
63 type_var: U
108 type_var: T1
113 type_var: T2
118 type_var: U
174 type_var: T1
184 type_var: U
[all …]
/llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/lang/
H A Dcomprehension.py147 self.type_var = type_var
330 def __call__(self, type_var: TypeVar, arg: TensorExpression) -> "TensorFn":
331 return TensorFn(FunctionKind.TYPE, self.fn_name, None, type_var, [arg])
426 type_var: Optional[TypeVar] = None,
431 if type_var and not isinstance(type_var, TypeVar):
435 self.type_var = type_var
489 type_var: TypeVar,
501 kind, type_var=type_var, size_exprs=shape, index_dims=index_dims)
544 def __init__(self, type_var: TypeVar):
545 self.operand_def = OperandDef(OperandKind.SCALAR, type_var=type_var)
[all …]
H A Dscalar_expr.py36 attr_name: Optional[str], type_var: Optional["TypeVar"],
43 self.type_var = type_var
126 if self.scalar_fn.type_var:
127 scalar_fn_dict["type_var"] = self.scalar_fn.type_var.name
H A Dconfig.py63 def type_var(self) -> TypeVar: member in OperandDefConfig
64 return self.operand_def.type_var
68 if self.type_var:
69 self_dict["type_var"] = self.type_var.name
H A Demitter.py303 operand_values = [expr.scalar_fn.type_var.name] + operand_values
515 name = operand_config.type_var.name
/llvm-project-15.0.7/mlir/test/mlir-linalg-ods-gen/
H A Dtest-linalg-ods-yaml-gen.yaml26 type_var: T
50 type_var: T
128 type_var: T
133 type_var: T
215 type_var: T1
219 type_var: U
233 type_var: U
269 type_var: T
352 type_var: T1
356 type_var: U
[all …]