| /external/llvm-project/mlir/include/mlir/Dialect/Affine/IR/ |
| D | AffineOps.h | 117 getSrcMap().getNumInputs()}; in getSrcIndices()
127 return getSrcMemRefOperandIndex() + 1 + getSrcMap().getNumInputs(); in getDstMemRefOperandIndex()
156 getDstMap().getNumInputs()}; in getDstIndices()
161 return getDstMemRefOperandIndex() + 1 + getDstMap().getNumInputs(); in getTagMemRefOperandIndex()
185 getTagMap().getNumInputs()}; in getTagIndices()
191 getTagMap().getNumInputs()); in getNumElements()
241 getTagMemRefOperandIndex() + 1 + getTagMap().getNumInputs() + 1; in isStrided()
298 operand_begin() + 1 + getTagMap().getNumInputs()}; in getTagIndices()
314 Value getNumElements() { return getOperand(1 + getTagMap().getNumInputs()); } in getNumElements()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/ |
| D | LinalgStructuredOpsInterface.td | 140 /*methodName=*/"getNumInputs"
170 assert(i < $_op.getNumInputs());
213 return {range.begin(), range.begin() + $_op.getNumInputs()};
266 return this->getOperation()->getOperand($_op.getNumInputs() + i);
306 return getShapedType(i + $_op.getNumInputs());
332 return {range.begin() + $_op.getNumInputs(),
375 return $_op.getNumInputs() + $_op.getNumOutputs();
388 return $_op.getNumInputs() + $_op.getNumOutputs() -
443 return i >= $_op.getNumInputs() + getNumOutputBuffers();
456 return i - $_op.getNumInputs() - getNumOutputBuffers();
[all …]
|
| D | LinalgTraits.h | 34 static unsigned getNumInputs() { return N; } in getNumInputs() function
98 unsigned getNumInputs() { in getNumInputs() function
|
| /external/llvm-project/mlir/lib/Dialect/Affine/IR/ |
| D | AffineOps.cpp | 581 assert(map.getNumInputs() == operands.size() && in AffineApplyNormalizer()
742 assert(mapOrSet->getNumInputs() == operands->size() && in canonicalizePromotedSymbols()
754 for (unsigned i = 0, e = mapOrSet->getNumInputs(); i != e; ++i) { in canonicalizePromotedSymbols()
774 assert(mapOrSet->getNumInputs() == operands->size() && in canonicalizePromotedSymbols()
788 assert(mapOrSet->getNumInputs() == operands->size() && in canonicalizeMapOrSetAndOperands()
1071 if (srcMapOperands.size() != srcMapAttr.getValue().getNumInputs() || in parse()
1072 dstMapOperands.size() != dstMapAttr.getValue().getNumInputs() || in parse()
1073 tagMapOperands.size() != tagMapAttr.getValue().getNumInputs()) in parse()
1091 unsigned numInputsAllMaps = getSrcMap().getNumInputs() + in verify()
1092 getDstMap().getNumInputs() + in verify()
[all …]
|
| /external/llvm-project/mlir/docs/ |
| D | Interfaces.md | 135 virtual unsigned getNumInputs(Operation *op) const = 0;
142 unsigned getNumInputs(Operation *op) const final {
143 return llvm::cast<OpT>(op).getNumInputs();
155 unsigned getNumInputs() const {
156 return getImpl()->getNumInputs(getOperation());
172 unsigned getNumInputs() { return ...; }
179 llvm::errs() << "num inputs = " << example.getNumInputs() << "\n";
200 "unsigned", "getNumInputs"
|
| /external/llvm-project/mlir/lib/IR/ |
| D | BuiltinDialect.cpp | 105 assert(type.getNumInputs() == argAttrs.size()); in build()
107 for (unsigned i = 0, e = type.getNumInputs(); i != e; ++i) in build()
178 inputTypes.reserve(newType.getNumInputs()); in clone()
|
| D | AffineMap.cpp | 216 unsigned AffineMap::getNumInputs() const { in getNumInputs() function in AffineMap
259 assert(getNumInputs() == operandConstants.size()); in partialConstantFold()
354 SmallVector<bool, 8> seen(getNumInputs(), false); in isProjectedPermutation()
436 if (seenExprs.size() != map.getNumInputs()) in inversePermutation()
|
| D | IntegerSet.cpp | 17 unsigned IntegerSet::getNumInputs() const { in getNumInputs() function in IntegerSet
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | FusionOnTensors.cpp | 73 for (unsigned argNum : llvm::seq<unsigned>(0, producer.getNumInputs())) { in getIndexingMapOfProducerOperandsInFusedOp()
185 producer.getNumInputs() + consumer.getNumInputs() - 1; in fuseTensorOpsImpl()
570 linalgOp.getIndexingMap(linalgOp.getNumInputs() + result.index()); in fuseWithReshapeByExpansion()
854 !isFusableWithReshapeByDimExpansion(producer, producer.getNumInputs())) in matchAndRewrite()
857 fuseWithReshapeByExpansion(producer, reshapeOp, producer.getNumInputs(), in matchAndRewrite()
918 linalgOp.getNumInputs() + operand.index(); in matchAndRewrite()
|
| D | Loops.cpp | 43 assert(map.getNumInputs() == vals.size()); in makeCanonicalAffineApplies()
144 unsigned nInputs = linalgOp.getNumInputs(); in emitScalarImplementation()
467 unsigned nInputs = indexedGenericOp.getNumInputs(); in emitScalarImplementation()
602 if (map.getNumResults() != 1 || map.getNumInputs() > 1) in matchAndRewrite()
606 if (map.getNumInputs() == 0) { in matchAndRewrite()
|
| D | Interchange.cpp | 47 if (linOp.getIndexingMap(0).getNumInputs() != interchangeVector.size()) in interchangeGenericLinalgOpPrecondition()
|
| /external/llvm-project/mlir/lib/Transforms/Utils/ |
| D | Utils.cpp | 59 assert(indexRemap.getNumInputs() == in replaceAllMemRefUsesWith()
103 unsigned oldMapNumInputs = oldMap.getNumInputs(); in replaceAllMemRefUsesWith()
233 assert(indexRemap.getNumInputs() == in replaceAllMemRefUsesWith()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | embed_tf_framework_pass.cc | 65 return func_type.getNumInputs() > 0 && in runOnOperation()
|
| D | embed_tf_framework.cc | 39 TypeConverter::SignatureConversion conversion(func_type.getNumInputs()); in matchAndRewrite()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | IntegerSet.h | 79 unsigned getNumInputs() const;
|
| D | AffineMap.h | 123 unsigned getNumInputs() const;
|
| /external/llvm-project/mlir/lib/Dialect/GPU/IR/ |
| D | GPUDialect.cpp | 528 return getBody().insertArgument(getType().getNumInputs() + attr.getInt(), in addWorkgroupAttribution()
638 unsigned numWorkgroupAttrs = argTypes.size() - type.getNumInputs(); in parseGPUFuncOp()
688 impl::printFunctionAttributes(p, op.getOperation(), type.getNumInputs(), in printGPUFuncOp()
701 for (int i = newType.getNumInputs(), e = oldType.getNumInputs(); i < e; i++) in setType()
|
| /external/llvm-project/mlir/lib/Conversion/GPUToSPIRV/ |
| D | ConvertGPUToSPIRV.cpp | 176 if (!argABIInfo.empty() && fnType.getNumInputs() != argABIInfo.size()) { in lowerAsEntryFunction()
185 TypeConverter::SignatureConversion signatureConverter(fnType.getNumInputs()); in lowerAsEntryFunction()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| D | LinalgOps.cpp | 410 auto nInputViews = op.getNumInputs(); in verify()
425 auto nInputViews = op.getNumInputs(); in verify()
1188 if (inputPermutationMap->getNumInputs() != rank) in verify()
1197 if (outputPermutationMap->getNumInputs() != rank) in verify()
1749 for (int i = 0, e = linalgOp.getNumInputs(); i != e; i++) { in matchAndRewrite()
1760 if (canonicalInput.size() == linalgOp.getNumInputs()) in matchAndRewrite()
1780 for (int i = 0, e = newLinalgOp.getNumInputs(); i != e; i++) in matchAndRewrite()
1797 bbArgBaseOffset = newIndexingMaps[0].getNumInputs(); in matchAndRewrite()
1802 for (int i = 0, e = linalgOp.getNumInputs(); i < e; i++) { in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/CAPI/IR/ |
| D | AffineMap.cpp | 98 return unwrap(affineMap).getNumInputs(); in mlirAffineMapGetNumInputs()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | mlir_tflite_runner.cc | 118 if (main.getType().getNumInputs() != 0) in main()
|
| /external/llvm-project/mlir/lib/Transforms/ |
| D | BufferResultsToOutParams.cpp | 44 func.setArgAttrs(functionType.getNumInputs() + i, in updateFuncOp()
|
| /external/llvm-project/mlir/lib/Dialect/StandardOps/Transforms/ |
| D | DecomposeCallGraphTypes.cpp | 69 TypeConverter::SignatureConversion conversion(functionType.getNumInputs()); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Conversion/LinalgToSPIRV/ |
| D | LinalgToSPIRV.cpp | 74 if (genericOp.getNumInputs() != 1 || genericOp.getNumOutputs() != 1) in matchAsPerformingReduction()
|
| /external/llvm-project/flang/lib/Lower/ |
| D | Mangler.cpp | 164 auto e = funTy.getNumInputs(); in mangleIntrinsicProcedure()
|