Searched refs:newInputs (Results 1 – 3 of 3) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Func/IR/ |
| H A D | FuncOps.cpp | 310 SmallVector<Type, 4> newInputs; in clone() local
311 newInputs.reserve(oldNumArgs); in clone()
314 newInputs.push_back(oldType.getInput(i)); in clone()
318 if (newInputs.size() != oldNumArgs) { in clone()
319 newFunc.setType(FunctionType::get(oldType.getContext(), newInputs, in clone()
324 newArgAttrs.reserve(newInputs.size()); in clone()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | SplitReduction.cpp | 116 SmallVector<Value> newInputs; in splitReduction() local
145 newInputs.push_back(operand->get()); in splitReduction()
153 newInputs.push_back(newInput); in splitReduction()
202 loc, TypeRange({initOrAllocTensor.getType()}), newInputs, in splitReduction()
377 auto newInputs = llvm::to_vector<4>(op.inputs()); in splitReductionByScaling() local
380 newInputs.push_back(b.create<linalg::InitTensorOp>( in splitReductionByScaling()
392 b.create<GenericOp>(loc, ValueRange(newOutputs).getTypes(), newInputs, in splitReductionByScaling()
|
| H A D | DropUnitDims.cpp | 423 SmallVector<Value, 4> newInputs = insertReshapes(genericOp.inputs()); in matchAndRewrite() local
433 loc, resultTypes, newInputs, newOutputs, newIndexingMaps, in matchAndRewrite()
|