Searched refs:TensorFn (Results 1 – 1 of 1) sorted by relevance
| /llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/lang/ |
| H A D | comprehension.py | 136 class TensorFn(TensorExpression): class
267 return TensorFn(FunctionKind.UNARY, self.fn_name, None, None, [arg])
295 return TensorFn(FunctionKind.BINARY, self.fn_name, None, None, [arg0, arg1])
331 return TensorFn(FunctionKind.TYPE, self.fn_name, None, type_var, [arg])
594 def __call__(self, arg: TensorExpression) -> TensorFn:
595 return TensorFn(FunctionKind.UNARY, None, self.operand_def, None, [arg])
614 arg1: TensorExpression) -> TensorFn:
615 return TensorFn(FunctionKind.BINARY, None, self.operand_def, None,
637 def __call__(self, type_var: TypeVar, arg: TensorExpression) -> TensorFn:
638 return TensorFn(FunctionKind.TYPE, None, self.operand_def, type_var, [arg])
|