Searched refs:bindDims (Results 1 – 7 of 7) sorted by relevance
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | AffineExpr.h | 279 template <int N> void bindDims(MLIRContext *ctx) {} in bindDims() function
282 void bindDims(MLIRContext *ctx, AffineExprTy &e, AffineExprTy2 &... exprs) { in bindDims() function
284 bindDims<N + 1, AffineExprTy2 &...>(ctx, exprs...); in bindDims()
291 void bindDims(MLIRContext *ctx, AffineExprTy &... exprs) { in bindDims() function
292 detail::bindDims<0>(ctx, exprs...); in bindDims()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Utils/ |
| D | StructuredOpsUtils.h | 30 bindDims(context, m, n, k); in isRowMajorMatmul()
41 bindDims(context, m, n, k); in isColumnMajorMatmul()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/EDSC/ |
| D | Builders.cpp | 179 bindDims(ScopedContext::getContext(), m, n, k); in linalg_generic_matmul()
197 bindDims(ScopedContext::getContext(), m, n, k); in linalg_generic_matmul()
225 bindDims(ctx, b, f, h, w, kh, kw, c); in linalg_generic_conv_nhwc()
262 bindDims(ctx, b, dm, c, h, w, kh, kw); in linalg_generic_dilated_conv_nhwc()
|
| /external/llvm-project/mlir/lib/Dialect/Vector/EDSC/ |
| D | Builders.cpp | 34 bindDims(ScopedContext::getContext(), m, n, k); in vector_contraction_matmul()
|
| /external/llvm-project/mlir/test/EDSC/ |
| D | builder-api-test.cpp | 929 bindDims(&globalContext(), i, j); in TEST_FUNC()
1058 bindDims(&globalContext(), i, j, k); in TEST_FUNC()
1124 bindDims(&globalContext(), i, j); in TEST_FUNC()
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorTransforms.cpp | 1707 bindDims(rewriter.getContext(), m, n, k); in matchAndRewrite()
1819 bindDims(rewriter.getContext(), m, n, k); in matchAndRewrite()
2230 bindDims(xferOp.getContext(), i, j, k); in createScopedSubViewIntersection()
|
| /external/llvm-project/mlir/docs/Dialects/ |
| D | Linalg.md | 663 bindDims(context, d0, d1, d2, d3);
|