| /external/llvm-project/mlir/lib/IR/ |
| D | AffineMap.cpp | 110 getMinorIdentityMap(getNumDims(), getNumResults(), getContext()); in isMinorIdentity()
180 if (getNumDims() != getNumResults()) in isIdentity()
192 return getNumDims() == 0 && getNumSymbols() == 0 && getNumResults() == 0; in isEmpty()
196 return getNumResults() == 1 && getResult(0).isa<AffineConstantExpr>(); in isSingleConstant()
212 unsigned AffineMap::getNumResults() const { in getNumResults() function in AffineMap
264 exprs.reserve(getNumResults()); in partialConstantFold()
304 results.reserve(getNumResults()); in replaceDimsAndSymbols()
313 assert(getNumDims() == map.getNumResults() && "Number of results mismatch"); in compose()
345 res.reserve(resMap.getNumResults()); in compose()
368 if (getNumDims() != getNumResults()) in isPermutation()
[all …]
|
| D | PatternMatch.cpp | 165 assert(op->getNumResults() == newValues.size() && in replaceOp()
244 assert(op->getNumResults() == newOp->getNumResults() && in replaceOpWithResultsOfAnotherOp()
246 if (op->getNumResults() == 1) in replaceOpWithResultsOfAnotherOp()
|
| D | Operation.cpp | 554 unsigned Operation::getNumResults() { in getNumResults() function in Operation
626 for (unsigned i = 0, e = getNumResults(); i != e; ++i) in cloneWithoutRegions()
833 if (op->getNumResults() != 0) in verifyZeroResult()
839 if (op->getNumResults() != 1) in verifyOneResult()
846 if (op->getNumResults() != numOperands) in verifyNResults()
853 if (op->getNumResults() < numOperands) in verifyAtLeastNResults()
1053 else if (!isOperand && totalCount != op->getNumResults()) in verifyValueSizeAttr()
1055 << op->getNumResults() << ") does not match with the total size (" in verifyValueSizeAttr()
1119 if (resultMappableTypes.size() != op->getNumResults()) in verifyElementwiseMappable()
1163 assert(op->getNumResults() == 1 && "op should have one result"); in printOneResultOp()
[all …]
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/ |
| D | LinalgTraits.h | 79 if (op->getNumResults() > concreteOp.getNumOutputs()) in verifyTrait()
121 if (redDims.empty() || op->getNumResults() == 0) { in verifyTrait()
129 if (op->getNumResults() != 1) in verifyTrait()
133 if (concreteOp.init_tensors().size() != op->getNumResults()) in verifyTrait()
137 for (unsigned idx = 0, e = op->getNumResults(); idx < e; ++idx) in verifyTrait()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | materialize_mlir_passthrough_op.cc | 66 if (main.getType().getNumResults() != op->getNumResults()) { in runOnFunction()
68 << op->getNumResults() in runOnFunction()
70 << main.getType().getNumResults() << " results)\n"; in runOnFunction()
|
| D | functional_control_flow_to_cfg.cc | 118 assert(op->getNumResults() == block->getNumArguments()); in ReplaceOpResultWithBlockArgs()
119 for (unsigned i = 0, e = op->getNumResults(); i != e; ++i) { in ReplaceOpResultWithBlockArgs()
249 assert(cond_call_op->getNumResults() == 1); in LowerWhileOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils_test.cc | 132 EXPECT_EQ(fused_lstm_func_.getType().getNumResults(), 1); in TEST_F()
160 EXPECT_EQ(it->getNumResults(), 1); in TEST_F()
182 EXPECT_EQ(fused_lstm_func_.getType().getNumResults(), 1); in TEST_F()
213 EXPECT_EQ(it->getNumResults(), 1); in TEST_F()
230 EXPECT_EQ(fused_ln_lstm_func_.getType().getNumResults(), 1); in TEST_F()
239 EXPECT_EQ(it->getNumResults(), 1); in TEST_F()
251 EXPECT_EQ(fused_ln_lstm_func_.getType().getNumResults(), 1); in TEST_F()
|
| D | perception_ops_utils.cc | 96 if (func_.getType().getNumResults() != 1) { in VerifySignature()
99 << func_.getType().getNumResults(); in VerifySignature()
169 if (func_.getType().getNumResults() != 1) { in VerifySignature()
172 << func_.getType().getNumResults(); in VerifySignature()
|
| D | nms_utils.cc | 70 if (func_.getType().getNumResults() != 2) { in VerifySignature()
73 << func_.getType().getNumResults(); in VerifySignature()
165 if (func_.getType().getNumResults() != 4) { in VerifySignature()
168 << ": " << func_.getType().getNumResults(); in VerifySignature()
|
| D | tftext_utils.cc | 103 if (func.getNumResults() != kValidNumOfOutput[input_type.getRank()]) { in VerifyWhitespaceTokenizer()
114 if (func.getNumResults() > 1) { in VerifyWhitespaceTokenizer()
121 if (func.getNumResults() > 2) { in VerifyWhitespaceTokenizer()
275 func.getType().getNumResults() != 1) { in VerifySgnnProjection()
|
| /external/llvm-project/mlir/lib/Dialect/Affine/IR/ |
| D | AffineValueMap.cpp | 28 assert(a.getNumResults() == b.getNumResults() && "invalid inputs"); in difference()
49 diffExprs.reserve(a.getNumResults()); in difference()
50 for (unsigned i = 0, e = bMap.getNumResults(); i < e; ++i) in difference()
|
| D | AffineOps.cpp | 375 result.types.append(map.getNumResults(), indexTy); in parseAffineApplyOp()
396 if (map.getNumResults() != 1) in verify()
1292 unsigned opNumResults = op.getNumResults(); in verify()
1376 if (map.getNumResults() > 1 && failedToParsedMinMax) { in parseBound()
1481 if (map.getNumResults() == 1) { in printBound()
1566 assert(boundMap.getNumResults() >= 1 && in foldLoopBounds()
1665 assert(map.getNumResults() >= 1 && "bound map has at least one result"); in setLowerBound()
1680 assert(map.getNumResults() >= 1 && "bound map has at least one result"); in setUpperBound()
1695 assert(map.getNumResults() >= 1 && "bound map has at least one result"); in setLowerBoundMap()
1704 assert(map.getNumResults() >= 1 && "bound map has at least one result"); in setUpperBoundMap()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_context.cc | 54 for (int res = 0, e = op.getNumResults(); res != e; ++res) { in QuantizeContext()
73 for (int i = 0; i < op.getNumResults(); ++i) { in GetSignature()
142 for (int res = 0, e = op.getNumResults(); res != e; ++res) { in Finalize()
173 for (auto i = 0; i < op.getNumResults(); ++i) { in DumpStates()
214 for (int i = 0, e = op->getNumResults(); i != e; ++i) { in GetQuantParamsForSameScaleConstraint()
226 if (op->getNumResults() == 1 && immutable_results_num == 1) { in GetQuantParamsForSameScaleConstraint()
245 if (op->getNumResults() == 1 && mutable_results_num == 1) { in GetQuantParamsForSameScaleConstraint()
269 for (int res = 0, e = op->getNumResults(); res != e; ++res) { in PropagateQuantParams()
|
| D | quantization_driver.cc | 330 for (int i = 0, e = op->getNumResults(); i < e; ++i) { in DumpStates()
403 for (int i = 0, e = op->getNumResults(); i != e; ++i) { in IsQuantized()
634 for (int i = 0, e = op->getNumResults(); i != e; ++i) { in GetQuantParamsForSameScaleConstraint()
646 if (op->getNumResults() == 1 && immutable_results_num == 1) { in GetQuantParamsForSameScaleConstraint()
661 if (op->getNumResults() == 1 && mutable_results_num == 1) { in GetQuantParamsForSameScaleConstraint()
762 for (int res = 0, e = op->getNumResults(); res != e; ++res) { in SetupAllStates()
838 for (int res = 0, e = op->getNumResults(); res != e; ++res) in PropagateParams()
854 for (auto i = 0; i < op->getNumResults(); ++i) { in PropagateParams()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | FunctionSupport.h | 246 unsigned getNumResults() { in getNumResults() function
281 unsigned originalNumResults = getNumResults(); in eraseResults()
368 for (unsigned i = 0, e = getNumResults(); i != e; ++i) in getAllResultAttrs()
396 assert(attributes.size() == getNumResults()); in setAllResultAttrs()
440 assert(index < getNumResults() && "invalid result number"); in getResultAttrDict()
494 for (unsigned i = 0, e = funcOp.getNumResults(); i != e; ++i) { in verifyTrait()
550 for (int i = newType.getNumResults(), e = oldType.getNumResults(); i < e; i++) in setType()
624 assert(index < getNumResults() && "invalid result number"); in setResultAttrs()
637 assert(index < getNumResults() && "invalid result number"); in setResultAttrs()
|
| D | Operation.h | 156 assert(std::distance(values.begin(), values.end()) == getNumResults() && in replaceAllUsesWith()
160 for (unsigned i = 0, e = getNumResults(); i != e; ++i) in replaceAllUsesWith()
166 assert(getNumResults() == op->getNumResults()); in replaceAllUsesWith()
167 for (unsigned i = 0, e = getNumResults(); i != e; ++i) in replaceAllUsesWith()
275 unsigned getNumResults();
665 unsigned numResults = getNumResults(); in prefixAllocSize()
|
| /external/llvm-project/mlir/tools/mlir-tblgen/ |
| D | OpPythonBindingGen.cpp | 365 static int getNumResults(const Operator &op) { return op.getNumResults(); } in getNumResults() function
384 emitElementAccessors(op, os, "result", getNumVariadic, getNumResults, in emitResultAccessors()
520 for (int i = 0, e = op.getNumResults(); i < e; ++i) { in populateBuilderArgs()
622 builderArgs.reserve(op.getNumOperands() + op.getNumResults() + in emitDefaultOpBuilder()
627 llvm::makeArrayRef(builderArgs).take_front(op.getNumResults()), in emitDefaultOpBuilder()
628 builderLines, getNumResults, getResult); in emitDefaultOpBuilder()
632 op, llvm::makeArrayRef(builderArgs).drop_front(op.getNumResults()), in emitDefaultOpBuilder()
|
| D | OpDefinitionsGen.cpp | 355 for (int i = 0, e = op.getNumResults(); i < e; ++i) { in populateSubstitutions()
769 const int numResults = op.getNumResults(); in genNamedResultGetters()
947 for (int i = 0, e = op.getNumResults(); i < e; ++i) { in genSeparateArgParamBuilder()
955 int numResults = op.getNumResults(); in genSeparateArgParamBuilder()
985 int numResults = op.getNumResults(); in genUseOperandAsResultTypeCollectiveParamBuilder()
1041 int numResults = op.getNumResults(); in genInferredTypeCollectiveParamBuilder()
1099 auto numResults = op.getNumResults(); in genUseOperandAsResultTypeSeparateParamBuilder()
1147 SmallVector<std::string, 2> resultTypes(op.getNumResults(), resultType); in genUseAttrAsResultTypeBuilder()
1294 int numResults = op.getNumResults(); in genCollectiveParamBuilder()
1358 auto numResults = op.getNumResults(); in buildParamList()
[all …]
|
| /external/llvm-project/mlir/lib/Transforms/ |
| D | NormalizeMemRefs.cpp | 167 llvm::seq<unsigned>(0, callOp.getNumResults())) { in areMemRefsNormalizable()
263 for (unsigned resIndex : llvm::seq<unsigned>(0, userOp->getNumResults())) { in updateFunctionSignature()
395 op->getNumResults() > 0 && !isa<CallOp>(op) && !funcOp.isExternal()) { in normalizeFuncOpMemRefs()
403 for (unsigned resIndex : llvm::seq<unsigned>(0, op->getNumResults())) { in normalizeFuncOpMemRefs()
449 llvm::seq<unsigned>(0, functionType.getNumResults())) { in normalizeFuncOpMemRefs()
491 for (unsigned resIndex : llvm::seq<unsigned>(0, oldOp->getNumResults())) { in createOpResultsNormalized()
|
| /external/llvm-project/mlir/lib/Analysis/ |
| D | LoopAnalysis.cpp | 53 if (lbMap.getNumResults() != 1) { in buildTripCountMapAndOperands()
63 SmallVector<AffineExpr, 4> lbSplatExpr(ubValueMap.getNumResults(), in buildTripCountMapAndOperands()
71 for (unsigned i = 0, e = tripCountValueMap.getNumResults(); i < e; ++i) in buildTripCountMapAndOperands()
123 assert(map.getNumResults() >= 1 && "expected one or more results"); in getLargestDivisorOfTripCount()
368 for (unsigned i = 0, e = op.getNumResults(); i < e; ++i) { in isOpwiseShiftValid()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Affine/IR/ |
| D | AffineValueMap.h | 71 inline unsigned getNumResults() const { return map.getNumResults(); } in getNumResults() function
|
| /external/llvm-project/mlir/lib/Dialect/Affine/Utils/ |
| D | Utils.cpp | 145 bool needsMax = lowerBoundMap.getNumResults() > 1; in affineParallelize()
146 bool needsMin = upperBoundMap.getNumResults() > 1; in affineParallelize()
183 if (ifOp.getNumResults() != 0) in hoistAffineIfOp()
|
| /external/llvm-project/mlir/lib/Transforms/Utils/ |
| D | Utils.cpp | 61 assert(indexRemap.getNumResults() + extraIndices.size() == newMemRefRank); in replaceAllMemRefUsesWith()
159 assert(extraIndex.getDefiningOp()->getNumResults() == 1 && in replaceAllMemRefUsesWith()
199 state.types.reserve(op->getNumResults()); in replaceAllMemRefUsesWith()
235 assert(indexRemap.getNumResults() + extraIndices.size() == newMemRefRank); in replaceAllMemRefUsesWith()
366 sliceOps->reserve(composedMap.getNumResults()); in createAffineComputationSlice()
470 unsigned newRank = layoutMap.getNumResults(); in normalizeMemRefType()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_executor.cc | 189 if (fetch.getNumOperands() < graph.getNumResults()) in Verify()
195 const int64_t num_results = graph.getNumResults(); in Verify()
272 return wrapped_op.getNumResults() == yield.getNumOperands() && in WrapsSingleOp()
289 int result_count = island.getNumResults() - 1; // -1 for the control token in Verify()
304 for (int operand_idx : llvm::seq<int>(0, island.getNumResults() - 1)) { in Verify()
469 if (switchn.getNumResults() != num_outs.getInt() + 1) in Verify()
472 << (switchn.getNumResults() - 1); in Verify()
520 p << " of " << (switchn.getNumResults() - 1); in Print()
1037 if (op.getNumOperands() != 0 || op.getNumResults() != 1 || in matchAndRewrite()
1060 if (op.getNumOperands() != 0 || op.getNumResults() != 2 || in matchAndRewrite()
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Hoisting.cpp | 63 if (op->getNumResults() > 0) { in hoistViewAllocOps()
64 assert(op->getNumResults() == 1 && "Unexpected multi-result alloc"); in hoistViewAllocOps()
|