Home
last modified time | relevance | path

Searched refs:initTensors (Results 1 – 6 of 6) sorted by relevance

/external/llvm-project/mlir/lib/Dialect/Linalg/EDSC/
DBuilders.cpp26 ArrayRef<StructuredIndexed> outputBuffers, ArrayRef<Value> initTensors, in makeGenericLinalgOp() argument
34 exprsList.reserve(inputs.size() + outputBuffers.size() + initTensors.size()); in makeGenericLinalgOp()
50 initTensorValues.reserve(initTensors.size()); in makeGenericLinalgOp()
54 std::copy(initTensors.begin(), initTensors.end(), in makeGenericLinalgOp()
80 blockTypes.reserve(inputs.size() + outputBuffers.size() + initTensors.size()); in makeGenericLinalgOp()
84 for (Value v : initTensors) in makeGenericLinalgOp()
/external/llvm-project/mlir/lib/Dialect/Linalg/IR/
DLinalgOps.cpp155 ValueRange inputs, ValueRange outputBuffers, ValueRange initTensors, in build() argument
159 build(builder, result, resultTensorTypes, inputs, outputBuffers, initTensors, in build()
169 for (ValueRange container : {inputs, outputBuffers, initTensors}) in build()
200 ValueRange inputs, ValueRange outputBuffers, ValueRange initTensors, in build() argument
203 build(builder, result, resultTensorTypes, inputs, outputBuffers, initTensors, in build()
210 ValueRange inputs, ValueRange outputBuffers, ValueRange initTensors, in build() argument
215 build(builder, result, resultTensorTypes, inputs, outputBuffers, initTensors, in build()
226 for (ValueRange container : {inputs, outputBuffers, initTensors}) in build()
260 ValueRange inputs, ValueRange outputBuffers, ValueRange initTensors, in build() argument
264 build(builder, result, resultTensorTypes, inputs, outputBuffers, initTensors, in build()
/external/llvm-project/mlir/include/mlir/Dialect/Linalg/EDSC/
DBuilders.h51 ArrayRef<StructuredIndexed> outputBuffers, ArrayRef<Value> initTensors,
/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DTiling.cpp377 auto initTensors = op.getInitTensors(); in tileLinalgOpImpl() local
379 loopRanges, /*iterArgInitValues*/ initTensors, iteratorTypes, in tileLinalgOpImpl()
/external/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/
DLinalgStructuredOps.td642 "ValueRange":$outputBuffers, "ValueRange":$initTensors,
651 "ValueRange":$outputBuffers, "ValueRange":$initTensors,
782 "ValueRange":$outputBuffers, "ValueRange":$initTensors,
793 "ValueRange":$outputBuffers, "ValueRange":$initTensors,
DLinalgStructuredOpsInterface.td128 // Num input/output/initTensors arguments handling.