Home
last modified time | relevance | path

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

/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DDropUnitDims.cpp329 auto insertReshapes = [&](ValueRange values) { in matchAndRewrite() local
344 SmallVector<Value, 4> newInputs = insertReshapes(op.inputs()); in matchAndRewrite()
346 insertReshapes(op.output_buffers()); in matchAndRewrite()
347 SmallVector<Value, 4> newInitTensors = insertReshapes(op.init_tensors()); in matchAndRewrite()