Home
last modified time | relevance | path

Searched refs:numDims (Results 1 – 25 of 39) sorted by relevance

12

/external/llvm-project/mlir/include/mlir/Analysis/
DAffineStructures.h66 unsigned numReservedCols, unsigned numDims = 0,
69 : numReservedCols(numReservedCols), numDims(numDims), in numReservedCols()
71 assert(numReservedCols >= numDims + numSymbols + 1); in numReservedCols()
72 assert(idArgs.empty() || idArgs.size() == numDims + numSymbols + numLocals); in numReservedCols()
75 numIds = numDims + numSymbols + numLocals; in numReservedCols()
85 FlatAffineConstraints(unsigned numDims = 0, unsigned numSymbols = 0,
88 : numReservedCols(numDims + numSymbols + numLocals + 1), numDims(numDims),
90 assert(numReservedCols >= numDims + numSymbols + 1);
91 assert(idArgs.empty() || idArgs.size() == numDims + numSymbols + numLocals);
92 numIds = numDims + numSymbols + numLocals;
[all …]
/external/llvm-project/mlir/lib/IR/
DAffineMap.cpp29 AffineExprConstantFolder(unsigned numDims, ArrayRef<Attribute> operandConsts) in AffineExprConstantFolder() argument
30 : numDims(numDims), operandConsts(operandConsts) {} in AffineExprConstantFolder()
66 if (auto attr = operandConsts[numDims + in constantFoldImpl()
86 unsigned numDims; member in __anond02650d10111::AffineExprConstantFolder
168 AffineMap AffineMap::getMultiDimIdentityMap(unsigned numDims, in getMultiDimIdentityMap() argument
171 dimExprs.reserve(numDims); in getMultiDimIdentityMap()
172 for (unsigned i = 0; i < numDims; ++i) in getMultiDimIdentityMap()
174 return get(/*dimCount=*/numDims, /*symbolCount=*/0, dimExprs, context); in getMultiDimIdentityMap()
183 for (unsigned i = 0, numDims = getNumDims(); i < numDims; ++i) { in isIdentity() local
206 return map->numDims; in getNumDims()
[all …]
DAffineExpr.cpp827 unsigned numDims, in getAffineExprFromFlatForm() argument
832 assert(flatExprs.size() - numDims - numSymbols - 1 == localExprs.size() && in getAffineExprFromFlatForm()
837 for (unsigned j = 0; j < numDims + numSymbols; j++) { in getAffineExprFromFlatForm()
840 auto id = j < numDims ? getAffineDimExpr(j, context) in getAffineExprFromFlatForm()
841 : getAffineSymbolExpr(j - numDims, context); in getAffineExprFromFlatForm()
846 for (unsigned j = numDims + numSymbols, e = flatExprs.size() - 1; j < e; in getAffineExprFromFlatForm()
850 auto term = localExprs[j - numDims - numSymbols] * flatExprs[j]; in getAffineExprFromFlatForm()
861 SimpleAffineExprFlattener::SimpleAffineExprFlattener(unsigned numDims, in SimpleAffineExprFlattener() argument
863 : numDims(numDims), numSymbols(numSymbols), numLocals(0) { in SimpleAffineExprFlattener()
939 floorDividend, numDims, numSymbols, localExprs, context); in visitModExpr()
[all …]
DAffineMapDetail.h24 unsigned numDims; member
DBuiltinTypes.cpp607 unsigned numDims = m.getNumDims(); in getStridesAndOffset() local
609 offset = simplifyAffineExpr(offset, numDims, numSymbols); in getStridesAndOffset()
611 stride = simplifyAffineExpr(stride, numDims, numSymbols); in getStridesAndOffset()
756 unsigned numDims = maps[0].getNumDims(), nSymbols = maps[0].getNumSymbols(); in makeCanonicalStridedLayoutExpr() local
776 return simplifyAffineExpr(expr, numDims, nSymbols); in makeCanonicalStridedLayoutExpr()
/external/llvm-project/mlir/lib/Parser/
DAffineParser.cpp53 AffineMap parseAffineMapRange(unsigned numDims, unsigned numSymbols);
55 IntegerSet parseIntegerSetConstraints(unsigned numDims, unsigned numSymbols);
59 unsigned &numDims);
67 ParseResult parseDimIdList(unsigned &numDims);
69 ParseResult parseDimAndOptionalSymbolIdList(unsigned &numDims,
475 ParseResult AffineParser::parseDimIdList(unsigned &numDims) { in parseDimIdList() argument
482 auto dimension = getAffineDimExpr(numDims++, getContext()); in parseDimIdList()
500 AffineParser::parseDimAndOptionalSymbolIdList(unsigned &numDims, in parseDimAndOptionalSymbolIdList() argument
502 if (parseDimIdList(numDims)) { in parseDimAndOptionalSymbolIdList()
515 unsigned numDims = 0, numSymbols = 0; in parseAffineMapOrIntegerSetInline() local
[all …]
DTypeParser.cpp227 unsigned numDims = dimensions.size(); in parseMemRefType() local
265 if (map.getNumDims() != numDims) { in parseMemRefType()
269 << " and affine map" << i + 1 << ": " << numDims in parseMemRefType()
272 numDims = map.getNumResults(); in parseMemRefType()
/external/tensorflow/tensorflow/go/
Dattrs.go157 numDims := make([]C.int, meta.list_size)
160 …C.TF_OperationGetAttrShapeList(op.c, cname, &dims[0], &numDims[0], C.int(meta.list_size), &storage…
166 numDim := numDims[i]
228 numDims := meta.total_size
230 if numDims < 0 {
233 if numDims == 0 {
236 dims := make([]C.int64_t, numDims)
237 …C.TF_OperationGetAttrShape(op.c, cname, (*C.int64_t)(unsafe.Pointer(&dims[0])), C.int(numDims), st…
Dtensor_handle.go76 n, err := th.numDims()
91 func (th *TensorHandle) numDims() (int, error) { func
/external/llvm-project/mlir/include/mlir/Dialect/SDBM/
DSDBM.h136 return 1 + numDims + numSymbols + numTemporaries; in getNumVariables()
144 return 1 + numDims + position; in getSymbolPosition()
149 return 1 + numDims + numSymbols + position; in getTemporaryPosition()
153 unsigned numDims; variable
/external/llvm-project/mlir/include/mlir/IR/
DAffineExprVisitor.h256 unsigned numDims; variable
271 SimpleAffineExprFlattener(unsigned numDims, unsigned numSymbols);
315 return numDims + numSymbols + numLocals + 1; in getNumCols()
318 inline unsigned getLocalVarStartIndex() const { return numDims + numSymbols; } in getLocalVarStartIndex()
319 inline unsigned getSymbolStartIndex() const { return numDims; } in getSymbolStartIndex()
DAffineMap.h66 static AffineMap getMultiDimIdentityMap(unsigned numDims,
233 unsigned getNumDims() const { return numDims; } in getNumDims()
234 void setNumDims(unsigned d) { numDims = d; } in setNumDims()
256 unsigned numDims; member
DIntegerSet.h54 static IntegerSet getEmptySet(unsigned numDims, unsigned numSymbols, in getEmptySet() argument
58 return get(numDims, numSymbols, one, true); in getEmptySet()
DAffineExpr.h242 unsigned numDims, unsigned numSymbols,
275 AffineExpr simplifyAffineExpr(AffineExpr expr, unsigned numDims,
/external/llvm-project/mlir/lib/Analysis/
DAffineStructures.cpp70 getFlattenedAffineExprs(ArrayRef<AffineExpr> exprs, unsigned numDims, in getFlattenedAffineExprs() argument
75 localVarCst->reset(numDims, numSymbols); in getFlattenedAffineExprs()
79 AffineExprFlattener flattener(numDims, numSymbols, exprs[0].getContext()); in getFlattenedAffineExprs()
103 mlir::getFlattenedAffineExpr(AffineExpr expr, unsigned numDims, in getFlattenedAffineExpr() argument
108 LogicalResult ret = ::getFlattenedAffineExprs({expr}, numDims, numSymbols, in getFlattenedAffineExpr()
148 numDims = other.getNumDimIds(); in FlatAffineConstraints()
178 numIds(set.getNumDims() + set.getNumSymbols()), numDims(set.getNumDims()), in FlatAffineConstraints()
218 numDims = newNumDims; in reset()
220 numIds = numDims + numSymbols + newNumLocals; in reset()
295 numDims++; in addId()
[all …]
DAffineAnalysis.cpp314 unsigned numDims = srcDomain.getNumDimIds() + dstDomain.getNumDimIds(); in initDependenceConstraints() local
317 unsigned numIds = numDims + numSymbols + numLocals; in initDependenceConstraints()
321 dependenceConstraints->reset(numIneq, numEq, numCols, numDims, numSymbols, in initDependenceConstraints()
480 unsigned numDims = dependenceDomain->getNumDimIds(); in addMemRefAccessConstraints() local
483 unsigned newLocalIdOffset = numDims + numSymbols + domNumLocalIds; in addMemRefAccessConstraints()
/external/llvm-project/mlir/lib/Dialect/Linalg/EDSC/
DBuilders.cpp226 unsigned numDims = c.cast<AffineDimExpr>().getPosition() + 1; in linalg_generic_conv_nhwc() local
235 simplifyAffineExpr(s[0] * h + d[0] * kh, numDims, 0), in linalg_generic_conv_nhwc()
236 simplifyAffineExpr(s[1] * w + d[1] * kw, numDims, 0), in linalg_generic_conv_nhwc()
263 unsigned numDims = kw.cast<AffineDimExpr>().getPosition() + 1; in linalg_generic_dilated_conv_nhwc() local
271 simplifyAffineExpr(s[0] * h + d[0] * kh, numDims, 0), in linalg_generic_dilated_conv_nhwc()
272 simplifyAffineExpr(s[1] * w + d[1] * kw, numDims, 0), in linalg_generic_dilated_conv_nhwc()
276 O({b, h, w, simplifyAffineExpr(c * depth_multiplier + dm, numDims, 0)})}, in linalg_generic_dilated_conv_nhwc()
/external/llvm-project/mlir/lib/Dialect/Affine/IR/
DAffineOps.cpp303 unsigned numDims, OpAsmPrinter &printer) { in printDimAndSymbolList() argument
305 printer << '(' << operands.take_front(numDims) << ')'; in printDimAndSymbolList()
306 if (operands.size() > numDims) in printDimAndSymbolList()
307 printer << '[' << operands.drop_front(numDims) << ']'; in printDimAndSymbolList()
313 unsigned &numDims) { in parseDimAndSymbolList() argument
318 numDims = opInfos.size(); in parseDimAndSymbolList()
335 unsigned numDims) { in verifyDimAndSymbolIdentifiers() argument
338 if (opIt++ < numDims) { in verifyDimAndSymbolIdentifiers()
362 unsigned numDims; in parseAffineApplyOp() local
364 parseDimAndSymbolList(parser, result.operands, numDims) || in parseAffineApplyOp()
[all …]
/external/llvm-project/mlir/lib/Dialect/Affine/EDSC/
DBuilders.cpp87 categorizeValueByAffineType(MLIRContext *context, Value val, unsigned &numDims, in categorizeValueByAffineType() argument
97 d = getAffineDimExpr(numDims++, context); in categorizeValueByAffineType()
107 unsigned numDims = 0, numSymbols = 0; in createBinaryIndexHandle() local
111 categorizeValueByAffineType(context, lhs, numDims, numSymbols); in createBinaryIndexHandle()
113 categorizeValueByAffineType(context, rhs, numDims, numSymbols); in createBinaryIndexHandle()
119 auto map = AffineMap::get(numDims, numSymbols, affCombiner(d0, d1)); in createBinaryIndexHandle()
/external/llvm-project/mlir/lib/Dialect/SDBM/
DSDBM.cpp212 result.numDims = maxDim + 1; in get()
430 if (matrixPos < numDims) in getSDBMExpressions()
432 return SDBMSymbolExpr::get(dialect, matrixPos - numDims); in getSDBMExpressions()
446 unsigned numTrueVariables = numDims + numSymbols; in getSDBMExpressions()
514 if (matrixPos < numDims) in print()
516 matrixPos -= numDims; in print()
/external/llvm-project/mlir/lib/Conversion/AffineToStandard/
DAffineToStandard.cpp225 auto numDims = affineMap.getNumDims(); in expandAffineMap() local
228 [numDims, &builder, loc, operands](AffineExpr expr) { in expandAffineMap()
230 operands.take_front(numDims), in expandAffineMap()
231 operands.drop_front(numDims)); in expandAffineMap()
418 auto numDims = integerSet.getNumDims(); in matchAndRewrite() local
420 operandsRef.take_front(numDims), in matchAndRewrite()
421 operandsRef.drop_front(numDims)); in matchAndRewrite()
/external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/
DEagerOperation.java83 long[] shape = new long[numDims(outputNativeHandle)]; in shape()
169 private static native int numDims(long handle); in numDims() method in EagerOperation
DEagerOperationBuilder.java254 private static native void setAttrShape(long opHandle, String name, long[] shape, int numDims); in setAttrShape() argument
257 long opHandle, String name, long[] shapes, int[] numDims); in setAttrShapeList() argument
/external/llvm-project/mlir/lib/CAPI/IR/
DAffineMap.cpp50 intptr_t numDims) { in mlirAffineMapMultiDimIdentityGet() argument
51 return wrap(AffineMap::getMultiDimIdentityMap(numDims, unwrap(ctx))); in mlirAffineMapMultiDimIdentityGet()
/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DVectorization.cpp570 unsigned numDims = mapping.size(); in matchAndRewrite() local
583 AffineMap::getMultiDimIdentityMap(numDims, context), in matchAndRewrite()
584 AffineMap::getMultiDimIdentityMap(numDims, context), in matchAndRewrite()
585 AffineMap::get(numDims, 0, {}, context)}; in matchAndRewrite()
587 std::vector<StringRef> iteratorTypes(numDims, "reduction"); in matchAndRewrite()

12