Home
last modified time | relevance | path

Searched refs:ScalarDef (Results 1 – 9 of 9) sorted by relevance

/llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/ops/
H A Dcore_named_ops.py81 AZp=ScalarDef(I32),
82 BZp=ScalarDef(I32),
138 AZp=ScalarDef(I32),
139 BZp=ScalarDef(I32),
323 IZp=ScalarDef(I32),
324 KZp=ScalarDef(I32),
515 IZp=ScalarDef(I32),
516 KZp=ScalarDef(I32),
905 def fill_rng_2d(min=ScalarDef(F64),
906 max=ScalarDef(F64),
[all …]
/llvm-project-15.0.7/mlir/test/python/dialects/linalg/opdsl/
H A Demit_fill.py15 def fill_poly(value=ScalarDef(T1), O=TensorDef(U, output=True)):
H A Darguments.py49 def fill(value=ScalarDef(T), O=TensorDef(T, S.M, S.K, output=True)):
H A Dassignments.py113 def fill(value=ScalarDef(T), O=TensorDef(T, S.M, S.K, output=True)):
/llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/lang/
H A Ddsl.py133 (TensorDef, ScalarDef, IndexAttrDef, UnaryFnAttrDef,
H A Dcomprehension.py74 if isinstance(expr, ScalarDef):
536 class ScalarDef(TensorExpression): class
/llvm-project-15.0.7/mlir/test/mlir-linalg-ods-gen/
H A Dtest-linalg-ods-yaml-gen.yaml194 # def test3(value=ScalarDef(T1),
326 # def test5(value=ScalarDef(T1), O=TensorDef(U, output=True)):
/llvm-project-15.0.7/mlir/docs/Dialects/Linalg/
H A DOpDSL.md96 def copy_and_scale(val=ScalarDef(T),
105 `ScalarDef`, which specifies the type of the scalar operand. The tensors are
309 def fill(value=ScalarDef(T1),
/llvm-project-15.0.7/llvm/lib/Target/AArch64/GISel/
H A DAArch64RegisterBankInfo.cpp702 auto ScalarDef = MRI.getVRegDef(ScalarReg); in getInstrMapping() local
706 onlyDefinesFP(*ScalarDef, MRI, TRI))) in getInstrMapping()