Home
last modified time | relevance | path

Searched refs:newInputs (Results 1 – 4 of 4) sorted by relevance

/llvm-project/mlir/lib/Dialect/Func/IR/
H A DFuncOps.cpp252 SmallVector<Type, 4> newInputs; in clone()
253 newInputs.reserve(oldNumArgs); in clone()
256 newInputs.push_back(oldType.getInput(i)); in clone()
260 if (newInputs.size() != oldNumArgs) { in clone()
261 newFunc.setType(FunctionType::get(oldType.getContext(), newInputs, in clone()
266 newArgAttrs.reserve(newInputs.size()); in clone()
251 SmallVector<Type, 4> newInputs; clone() local
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DTransposeConv2D.cpp96 SmallVector<Value> newInputs{op.getInputs()}; in transposeConv2DHelper() local
99 newInputs[1] = newFilter; in transposeConv2DHelper()
107 rewriter.create<HWCFConvOp>(loc, resultTy, newInputs, op.getOutputs(), in transposeConv2DHelper()
H A DSplitReduction.cpp74 SmallVector<Value> newInputs; in splitReduction() local
111 newInputs.push_back(operand->get()); in splitReduction()
120 newInputs.push_back(newInput); in splitReduction()
172 loc, TypeRange({emptyOrAllocTensor.getType()}), newInputs, in splitReduction()
354 SmallVector<Value> newInputs = op.getDpsInputs(); in splitReductionByScaling() local
357 newInputs.push_back(b.create<tensor::EmptyOp>( in splitReductionByScaling()
368 b.create<GenericOp>(loc, ValueRange(newOutputs).getTypes(), newInputs, in splitReductionByScaling()
H A DDropUnitDims.cpp513 ArrayRef<Value> newInputs = in dropUnitDims()
522 rewriter.create<GenericOp>(loc, resultTypes, newInputs, newOutputs, in dropUnitDims()
514 ArrayRef<Value> newInputs = dropUnitDims() local