Home
last modified time | relevance | path

Searched refs:IndexAttrDef (Results 1 – 8 of 8) sorted by relevance

/llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/ops/
H A Dcore_named_ops.py259 strides=IndexAttrDef(S.SW, default=[1]),
397 strides=IndexAttrDef(S.SD,
401 dilations=IndexAttrDef(S.DD,
469 dilations=IndexAttrDef(S.DH,
496 dilations=IndexAttrDef(S.DH,
523 strides=IndexAttrDef(S.SH,
556 dilations=IndexAttrDef(S.DH,
613 strides=IndexAttrDef(S.SD,
819 dilations=IndexAttrDef(S.DD,
846 dilations=IndexAttrDef(S.DD,
[all …]
/llvm-project-15.0.7/mlir/test/python/dialects/linalg/opdsl/
H A Demit_convolution.py19 strides=IndexAttrDef(S.SH, S.SW, default=[1, 1]),
20 dilations=IndexAttrDef(S.DH, S.DW, default=[1, 2])):
H A Demit_pooling.py21 strides=IndexAttrDef(S.SH, S.SW, default=[1, 1]),
22 dilations=IndexAttrDef(S.DH, S.DW, default=[1, 1])):
H A Darguments.py74 strides=IndexAttrDef(S.SH, S.SW, default=[1, 2])):
/llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/lang/
H A Ddsl.py133 (TensorDef, ScalarDef, IndexAttrDef, UnaryFnAttrDef,
H A Dcomprehension.py557 class IndexAttrDef: class
/llvm-project-15.0.7/mlir/docs/Dialects/Linalg/
H A DOpDSL.md126 strides=IndexAttrDef(S.SH, S.SW, default=[1, 1])):
132 tensor `O`. The `strides` attribute is bound to an `IndexAttrDef`. It defines
166 strides=IndexAttrDef(S.SH, S.SW, default=[1, 1]),
167 dilations=IndexAttrDef(S.DH, S.DW, default=[1, 1])):
/llvm-project-15.0.7/mlir/test/mlir-linalg-ods-gen/
H A Dtest-linalg-ods-yaml-gen.yaml108 # strides=IndexAttrDef(S.SM, S.SN, default=[1, 2])):