Home
last modified time | relevance | path

Searched defs:numDims (Results 1 – 25 of 34) sorted by relevance

12

/external/llvm-project/mlir/lib/Parser/
DAffineParser.cpp475 ParseResult AffineParser::parseDimIdList(unsigned &numDims) { in parseDimIdList()
500 AffineParser::parseDimAndOptionalSymbolIdList(unsigned &numDims, in parseDimAndOptionalSymbolIdList()
515 unsigned numDims = 0, numSymbols = 0; in parseAffineMapOrIntegerSetInline() local
588 AffineMap AffineParser::parseAffineMapRange(unsigned numDims, in parseAffineMapRange()
656 IntegerSet AffineParser::parseIntegerSetConstraints(unsigned numDims, in parseIntegerSetConstraints()
DTypeParser.cpp227 unsigned numDims = dimensions.size(); in parseMemRefType() local
/external/llvm-project/mlir/lib/IR/
DAffineMap.cpp29 AffineExprConstantFolder(unsigned numDims, ArrayRef<Attribute> operandConsts) in AffineExprConstantFolder()
86 unsigned numDims; member in __anond02650d10111::AffineExprConstantFolder
168 AffineMap AffineMap::getMultiDimIdentityMap(unsigned numDims, in getMultiDimIdentityMap()
183 for (unsigned i = 0, numDims = getNumDims(); i < numDims; ++i) { in isIdentity() local
315 unsigned numDims = map.getNumDims(); in compose() local
442 unsigned numResults = 0, numDims = 0, numSymbols = 0; in concatAffineMaps() local
DAffineMapDetail.h24 unsigned numDims; member
DAffineExpr.cpp827 unsigned numDims, in getAffineExprFromFlatForm()
861 SimpleAffineExprFlattener::SimpleAffineExprFlattener(unsigned numDims, in SimpleAffineExprFlattener()
1072 AffineExpr mlir::simplifyAffineExpr(AffineExpr expr, unsigned numDims, in simplifyAffineExpr()
DBuiltinTypes.cpp607 unsigned numDims = m.getNumDims(); in getStridesAndOffset() local
756 unsigned numDims = maps[0].getNumDims(), nSymbols = maps[0].getNumSymbols(); in makeCanonicalStridedLayoutExpr() local
/external/llvm-project/mlir/include/mlir/IR/
DIntegerSet.h54 static IntegerSet getEmptySet(unsigned numDims, unsigned numSymbols, in getEmptySet()
DAffineMap.h256 unsigned numDims; member
DAffineExprVisitor.h256 unsigned numDims; variable
/external/tensorflow/tensorflow/go/
Dtensor_handle.go91 func (th *TensorHandle) numDims() (int, error) { func
/external/llvm-project/mlir/include/mlir/Dialect/SDBM/
DSDBM.h153 unsigned numDims; variable
/external/llvm-project/mlir/lib/Dialect/Affine/IR/
DAffineOps.cpp303 unsigned numDims, OpAsmPrinter &printer) { in printDimAndSymbolList()
313 unsigned &numDims) { in parseDimAndSymbolList()
335 unsigned numDims) { in verifyDimAndSymbolIdentifiers()
362 unsigned numDims; in parseAffineApplyOp() local
528 unsigned numDims = map.getNumDims(); in promoteComposedSymbolsAsDims() local
672 auto numDims = dimValueToPosition.size(); in AffineApplyNormalizer() local
1357 unsigned numDims; in parseBound() local
1929 unsigned numDims; in parseAffineIfOp() local
2293 unsigned numDims = op.map().getNumDims(); in printAffineMinMaxOp() local
2516 auto numDims = lbMap.getNumResults(); in build() local
[all …]
/external/llvm-project/mlir/lib/Dialect/Linalg/EDSC/
DBuilders.cpp226 unsigned numDims = c.cast<AffineDimExpr>().getPosition() + 1; in linalg_generic_conv_nhwc() local
263 unsigned numDims = kw.cast<AffineDimExpr>().getPosition() + 1; in linalg_generic_dilated_conv_nhwc() local
/external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/
DEagerOperationBuilder.java254 private static native void setAttrShape(long opHandle, String name, long[] shape, int numDims); in setAttrShape()
257 long opHandle, String name, long[] shapes, int[] numDims); in setAttrShapeList()
DEagerOperation.java169 private static native int numDims(long handle); in numDims() method in EagerOperation
DGraphOperationBuilder.java338 private static native void setAttrShape(long handle, String name, long[] shape, int numDims); in setAttrShape()
341 long handle, String name, long[] shapes, int[] numDims); in setAttrShapeList()
/external/llvm-project/mlir/lib/CAPI/IR/
DAffineMap.cpp50 intptr_t numDims) { in mlirAffineMapMultiDimIdentityGet()
/external/llvm-project/mlir/lib/Dialect/Affine/EDSC/
DBuilders.cpp87 categorizeValueByAffineType(MLIRContext *context, Value val, unsigned &numDims, in categorizeValueByAffineType()
107 unsigned numDims = 0, numSymbols = 0; in createBinaryIndexHandle() local
/external/llvm-project/mlir/include/mlir/Analysis/
DAffineStructures.h645 unsigned numDims; variable
/external/llvm-project/mlir/lib/Conversion/AffineToStandard/
DAffineToStandard.cpp225 auto numDims = affineMap.getNumDims(); in expandAffineMap() local
418 auto numDims = integerSet.getNumDims(); in matchAndRewrite() local
/external/llvm-project/mlir/lib/Analysis/
DLoopAnalysis.cpp239 unsigned numDims = accessMap.getNumDims(); in isContiguousAccess() local
DAffineStructures.cpp70 getFlattenedAffineExprs(ArrayRef<AffineExpr> exprs, unsigned numDims, in getFlattenedAffineExprs()
103 mlir::getFlattenedAffineExpr(AffineExpr expr, unsigned numDims, in getFlattenedAffineExpr()
2460 unsigned numDims = cst.getNumDimIds(); in getNewNumDimsSymbols() local
2919 unsigned numDims = cst.getNumDimIds(); in computeLocalVars() local
2949 unsigned numDims = getNumDimIds(); in getIneqAsAffineValueMap() local
3025 unsigned numDims = getNumDimIds(); in getAsIntegerSet() local
DAffineAnalysis.cpp314 unsigned numDims = srcDomain.getNumDimIds() + dstDomain.getNumDimIds(); in initDependenceConstraints() local
480 unsigned numDims = dependenceDomain->getNumDimIds(); in addMemRefAccessConstraints() local
DUtils.cpp69 unsigned numDims = ivs.size(); in getAsConstraints() local
269 unsigned numDims = accessMap.getNumDims(); in compute() local
/external/llvm-project/mlir/lib/Conversion/SCFToGPU/
DSCFToGPU.cpp89 unsigned numDims) { in checkAffineLoopNestMappableImpl()

12