Home
last modified time | relevance | path

Searched defs:kw (Results 1 – 13 of 13) sorted by relevance

/llvm-project/polly/lib/External/isl/imath/tests/gmp-compat-test/
H A Dgmpapi.py15 def __init__(self, ret_ty, name, *params, **kw):
/llvm-project/flang/unittests/Evaluate/
H A Dintrinsics.cpp55 template <typename A> static NamedArg<A> Named(std::string kw, A &&x) { in Named()
82 for (auto &kw : keywords) { in Marshal() local
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DVectorization.cpp91 for (int64_t kw = 0; kw < kwSize; ++kw) { in extractConvInputSlices() local
102 for (int64_t kw = 0; kw < kwSize; ++kw) { in extractConvInputSlices() local
122 for (int64_t kw = 0; kw < kwSize; ++kw) { extractConvFilterSlices() local
3048 __anon729377362d02(int64_t kw, int64_t w) conv() argument
3055 for (int64_t kw = 0; kw < kwSize; ++kw) { conv() local
3280 for (int64_t kw = 0; kw < kwSize; ++kw) { depthwiseConv() local
3289 for (int64_t kw = 0; kw < kwSize; ++kw) { depthwiseConv() local
3302 __anon729377362f02(int64_t kw, int64_t w) depthwiseConv() argument
3316 for (int64_t kw = 0; kw < kwSize; ++kw) { depthwiseConv() local
3420 AffineExpr w, kw; generateNonChanneledConv() local
3438 AffineExpr n, w, f, kw, c; generateNwcConv() local
3456 AffineExpr n, w, f, kw, c; generateNcwConv() local
3473 AffineExpr n, w, c, kw; generateNwcPooling() local
3491 AffineExpr n, w, c, kw; generateNcwPooling() local
3510 AffineExpr n, w, c, kw; generateDilatedConv() local
[all...]
/llvm-project/mlir/test/python/dialects/linalg/opdsl/
H A Demit_pooling.py17 K=TensorDef(T2, S.KH, S.KW, index_dims=[D.kh, D.kw]),
/llvm-project/flang/lib/Evaluate/
H A Dformatting.cpp163 EmitVar(llvm::raw_ostream & o,const A * p,const char * kw=nullptr) EmitVar() argument
175 EmitVar(llvm::raw_ostream & o,const std::optional<A> & x,const char * kw=nullptr) EmitVar() argument
187 EmitVar(llvm::raw_ostream & o,const common::Indirection<A,COPY> & p,const char * kw=nullptr) EmitVar() argument
H A Dintrinsics.cpp1763 std::string kw{d.keyword}; Match() local
2433 std::string kw{d.keyword}; Match() local
3198 if (auto kw{a->keyword()}; kw && kw == "dim") { Probe() local
[all...]
/llvm-project/flang/lib/Parser/
H A Dparse-tree.cpp170 if (auto &kw{std::get<std::optional<Keyword>>(arg.t)}) { in ConvertToStructureConstructor() local
/llvm-project/mlir/python/mlir/dialects/linalg/opdsl/ops/
H A Dcore_named_ops.py1388 pooling_nhwc_sum( I=TensorDef(T1, S.N, S.OH * S.SH + S.KH * S.DH, S.OW * S.SW + S.KW * S.DW, S.C), K=TensorDef(T2, S.KH, S.KW, index_dims=[D.kh, D.kw]), O=TensorDef(U, S.N, S.OH, S.OW, S.C, output=True), strides=IndexAttrDef(S.SH, S.SW, default=[1, 1]), dilations=IndexAttrDef(S.DH, S.DW, default=[1, 1]), ) global() argument
1412 pooling_nchw_sum( I=TensorDef(T1, S.N, S.C, S.OH * S.SH + S.KH * S.DH, S.OW * S.SW + S.KW * S.DW), K=TensorDef(T2, S.KH, S.KW, index_dims=[D.kh, D.kw]), O=TensorDef(U, S.N, S.C, S.OH, S.OW, output=True), strides=IndexAttrDef(S.SH, S.SW, default=[1, 1]), dilations=IndexAttrDef(S.DH, S.DW, default=[1, 1]), ) global() argument
1436 pooling_nhwc_max( I=TensorDef(T1, S.N, S.OH * S.SH + S.KH * S.DH, S.OW * S.SW + S.KW * S.DW, S.C), K=TensorDef(T2, S.KH, S.KW, index_dims=[D.kh, D.kw]), O=TensorDef(U, S.N, S.OH, S.OW, S.C, output=True), strides=IndexAttrDef(S.SH, S.SW, default=[1, 1]), dilations=IndexAttrDef(S.DH, S.DW, default=[1, 1]), ) global() argument
1458 pooling_nhwc_max_unsigned( I=TensorDef(T1, S.N, S.OH * S.SH + S.KH * S.DH, S.OW * S.SW + S.KW * S.DW, S.C), K=TensorDef(T2, S.KH, S.KW, index_dims=[D.kh, D.kw]), O=TensorDef(U, S.N, S.OH, S.OW, S.C, output=True), strides=IndexAttrDef(S.SH, S.SW, default=[1, 1]), dilations=IndexAttrDef(S.DH, S.DW, default=[1, 1]), ) global() argument
1480 pooling_nchw_max( I=TensorDef(T1, S.N, S.C, S.OH * S.SH + S.KH * S.DH, S.OW * S.SW + S.KW * S.DW), K=TensorDef(T2, S.KH, S.KW, index_dims=[D.kh, D.kw]), O=TensorDef(U, S.N, S.C, S.OH, S.OW, output=True), strides=IndexAttrDef(S.SH, S.SW, default=[1, 1]), dilations=IndexAttrDef(S.DH, S.DW, default=[1, 1]), ) global() argument
1508 pooling_nhwc_min( I=TensorDef(T1, S.N, S.OH * S.SH + S.KH * S.DH, S.OW * S.SW + S.KW * S.DW, S.C), K=TensorDef(T2, S.KH, S.KW, index_dims=[D.kh, D.kw]), O=TensorDef(U, S.N, S.OH, S.OW, S.C, output=True), strides=IndexAttrDef(S.SH, S.SW, default=[1, 1]), dilations=IndexAttrDef(S.DH, S.DW, default=[1, 1]), ) global() argument
1530 pooling_nhwc_min_unsigned( I=TensorDef(T1, S.N, S.OH * S.SH + S.KH * S.DH, S.OW * S.SW + S.KW * S.DW, S.C), K=TensorDef(T2, S.KH, S.KW, index_dims=[D.kh, D.kw]), O=TensorDef(U, S.N, S.OH, S.OW, S.C, output=True), strides=IndexAttrDef(S.SH, S.SW, default=[1, 1]), dilations=IndexAttrDef(S.DH, S.DW, default=[1, 1]), ) global() argument
1710 pooling_ndhwc_sum( I=TensorDef( T1, S.N, S.OD * S.SD + S.KD * S.DD, S.OH * S.SH + S.KH * S.DH, S.OW * S.SW + S.KW * S.DW, S.C, ), K=TensorDef(T2, S.KD, S.KH, S.KW, index_dims=[D.kd, D.kh, D.kw]), O=TensorDef(U, S.N, S.OD, S.OH, S.OW, S.C, output=True), strides=IndexAttrDef(S.SD, S.SH, S.SW, default=[1, 1, 1]), dilations=IndexAttrDef(S.DD, S.DH, S.DW, default=[1, 1, 1]), ) global() argument
1744 pooling_ndhwc_max( I=TensorDef( T1, S.N, S.OD * S.SD + S.KD * S.DD, S.OH * S.SH + S.KH * S.DH, S.OW * S.SW + S.KW * S.DW, S.C, ), K=TensorDef(T2, S.KD, S.KH, S.KW, index_dims=[D.kd, D.kh, D.kw]), O=TensorDef(U, S.N, S.OD, S.OH, S.OW, S.C, output=True), strides=IndexAttrDef(S.SD, S.SH, S.SW, default=[1, 1, 1]), dilations=IndexAttrDef(S.DD, S.DH, S.DW, default=[1, 1, 1]), ) global() argument
1780 pooling_ndhwc_min( I=TensorDef( T1, S.N, S.OD * S.SD + S.KD * S.DD, S.OH * S.SH + S.KH * S.DH, S.OW * S.SW + S.KW * S.DW, S.C, ), K=TensorDef(T2, S.KD, S.KH, S.KW, index_dims=[D.kd, D.kh, D.kw]), O=TensorDef(U, S.N, S.OD, S.OH, S.OW, S.C, output=True), strides=IndexAttrDef(S.SD, S.SH, S.SW, default=[1, 1, 1]), dilations=IndexAttrDef(S.DD, S.DH, S.DW, default=[1, 1, 1]), ) global() argument
[all...]
/llvm-project/flang/lib/Semantics/
H A Dcheck-call.cpp33 if (auto kw{arg.keyword()}) { in CheckImplicitInterfaceArg() local
H A Dexpression.cpp2029 if (const auto &kw{std::get<std::optional<parser::Keyword>>(component.t)}) { Analyze() local
H A Dresolve-names.cpp6079 if (const auto &kw{std::get<std::optional<parser::Keyword>>(component.t)}) { Pre() local
/llvm-project/mlir/lib/Dialect/LLVMIR/IR/
H A DLLVMDialect.cpp86 StringRef kw; parseOverflowFlags() local
/llvm-project/clang/lib/Sema/
H A DSemaLookup.cpp5025 for (auto *kw : CastableTypeSpecs) AddKeywordsToConsumer() local