Home
last modified time | relevance | path

Searched refs:parentPos (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/ExecutionEngine/
H A DSparseTensorUtils.cpp865 assert(parentPos < src.values.size() && in forallElements()
981 uint64_t parentPos = 0; in add() local
984 nnz[r][parentPos]++; in add()
985 parentPos = parentPos * dimSizes[r] + ind[r]; in add()
995 yield(nnz[d][parentPos]); in forallIndices()
1053 uint64_t parentSz = 1, parentPos = 0; in SparseTensorStorage() local
1066 pointers[r][parentPos]++; in SparseTensorStorage()
1068 parentPos = currentPos; in SparseTensorStorage()
1070 parentPos = parentPos * getDimSizes()[r] + ind[r]; in SparseTensorStorage()
1075 values[parentPos] = val; in SparseTensorStorage()
[all …]
/llvm-project-15.0.7/mlir/lib/Conversion/PDLToPDLInterp/
H A DPredicateTree.cpp91 OperationPosition *parentPos = builder.getOperandDefiningOp(pos); in getOperandTreePredicates() local
92 predList.emplace_back(parentPos, builder.getIsNotNull()); in getOperandTreePredicates()
98 resultPos = builder.getResult(parentPos, *index); in getOperandTreePredicates()
100 resultPos = builder.getResultGroup(parentPos, index, isVariadic); in getOperandTreePredicates()
105 (Position *)parentPos); in getOperandTreePredicates()
289 auto *parentPos = cast<OperationPosition>(inputs.lookup(op.parent())); in getResultPredicates() local
290 resultPos = builder.getResult(parentPos, op.index()); in getResultPredicates()
303 auto *parentPos = cast<OperationPosition>(inputs.lookup(op.parent())); in getResultPredicates() local
306 resultPos = builder.getResultGroup(parentPos, index, isVariadic); in getResultPredicates()