Home
last modified time | relevance | path

Searched refs:replaceOp (Results 1 – 25 of 101) sorted by relevance

12345

/external/llvm-project/mlir/lib/Conversion/StandardToSPIRV/
DLegalizeStandardForSPIRV.cpp36 void replaceOp(OpTy loadOp, SubViewOp subViewOp,
51 void replaceOp(OpTy StoreOp, SubViewOp subViewOp,
57 void LoadOpOfSubViewFolder<LoadOp>::replaceOp(LoadOp loadOp, in replaceOp() function in __anon615d5a650111::LoadOpOfSubViewFolder
66 void LoadOpOfSubViewFolder<vector::TransferReadOp>::replaceOp( in replaceOp() function in __anon615d5a650111::LoadOpOfSubViewFolder::TransferReadOp
75 void StoreOpOfSubViewFolder<StoreOp>::replaceOp( in replaceOp() function in __anon615d5a650111::StoreOpOfSubViewFolder
83 void StoreOpOfSubViewFolder<vector::TransferWriteOp>::replaceOp( in replaceOp() function in __anon615d5a650111::StoreOpOfSubViewFolder::TransferWriteOp
153 replaceOp(loadOp, subViewOp, sourceIndices, rewriter); in matchAndRewrite()
174 replaceOp(storeOp, subViewOp, sourceIndices, rewriter); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tfl.cc196 rewriter.replaceOp(op, {output}); in matchAndRewrite()
253 rewriter.replaceOp(op, {output}); in matchAndRewrite()
316 rewriter.replaceOp(op, {output}); in matchAndRewrite()
385 rewriter.replaceOp(op, {output}); in matchAndRewrite()
449 rewriter.replaceOp(op, {output}); in matchAndRewrite()
514 rewriter.replaceOp(op, {output}); in matchAndRewrite()
609 rewriter.replaceOp(op, {fused_activation_val.getValue()}); in matchAndRewrite()
613 rewriter.replaceOp(op, {output}); in matchAndRewrite()
706 rewriter.replaceOp(op, {fused_activation_val.getValue()}); in matchAndRewrite()
710 rewriter.replaceOp(op, {output}); in matchAndRewrite()
[all …]
Dlegalize_tf.cc209 rewriter.replaceOp(op, {op2_not_op1.getResult()}); in matchAndRewrite()
291 rewriter.replaceOp(op, {newOp.getResult()}); in matchAndRewrite()
319 rewriter.replaceOp(op, {result.getValue()}); in matchAndRewrite()
333 rewriter.replaceOp(op, {result.getValue()}); in matchAndRewrite()
347 rewriter.replaceOp(op, {result.getValue()}); in matchAndRewrite()
367 rewriter.replaceOp(op, {result.getValue()}); in matchAndRewrite()
386 rewriter.replaceOp(op, {result.getValue()}); in matchAndRewrite()
401 rewriter.replaceOp(op, {result.getValue()}); in matchAndRewrite()
457 rewriter.replaceOp(op, {mul_op.getResult()}); in matchAndRewrite()
649 rewriter.replaceOp(op, {result.getValue()}); in matchAndRewrite()
[all …]
/external/llvm-project/mlir/lib/Conversion/VectorToSPIRV/
DVectorToSPIRV.cpp42 rewriter.replaceOp(broadcastOp, construct); in matchAndRewrite()
60 rewriter.replaceOp(extractOp, newExtract); in matchAndRewrite()
77 rewriter.replaceOp(insertOp, newInsert); in matchAndRewrite()
95 rewriter.replaceOp(extractElementOp, newExtractElement); in matchAndRewrite()
113 rewriter.replaceOp(insertElementOp, newInsertElement); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dgpu_fusion.cc101 rewriter.replaceOp(batch_norm, op->getResults()); in matchAndRewrite()
108 rewriter.replaceOp(relu_op, op->getResult(0)); in matchAndRewrite()
111 rewriter.replaceOp(add_op, op->getResult(0)); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dtest_infer_shaped_type_pass.cc61 rewriter.replaceOp(op, {new_op->getResults()}); in matchAndRewrite()
79 rewriter.replaceOp(op, return_shapes); in matchAndRewrite()
Dhlo_legalize_to_lhlo.cc130 rewriter.replaceOp( in matchAndRewrite()
165 rewriter.replaceOp(op, ArrayRef<Value>(buffer_args).slice(operands.size())); in matchAndRewrite()
191 rewriter.replaceOp(op, ArrayRef<Value>(buffer_args).slice(operands.size())); in matchAndRewrite()
268 rewriter.replaceOp(op, {result}); in matchAndRewrite()
401 rewriter.replaceOp(op, bufferArgs[2]); in matchAndRewrite()
448 rewriter.replaceOp(op, ArrayRef<Value>(buffer_args).slice(operands.size())); in matchAndRewrite()
Dlegalize_to_linalg.cc223 rewriter.replaceOp(op, linalg_op.getOperation()->getResults()); in matchAndRewrite()
398 rewriter.replaceOp(op, linalg_op.getOperation()->getResults()); in matchAndRewrite()
527 rewriter.replaceOp(op, generic->getResults()); in matchAndRewrite()
582 rewriter.replaceOp(op, llvm::None); in matchAndRewrite()
879 rewriter.replaceOp(iota_op, llvm::None); in matchAndRewrite()
881 rewriter.replaceOp(iota_op, linalg_op.result_tensors()); in matchAndRewrite()
906 rewriter.replaceOp(op, {std_tensor_const}); in ReplaceConstOp()
997 rewriter.replaceOp(entry_block->getTerminator(), {}); in matchAndRewrite()
1005 rewriter.replaceOp(reduce_op, linalg_op.getOperation()->getResults()); in matchAndRewrite()
1174 rewriter.replaceOp(op, linalg_op->getResults()); in matchAndRewrite()
[all …]
/external/llvm-project/mlir/lib/Conversion/SCFToStandard/
DSCFToStandard.cpp344 rewriter.replaceOp(forOp, conditionBlock->getArguments().drop_front()); in matchAndRewrite()
399 rewriter.replaceOp(ifOp, continueBlock->getArguments()); in matchAndRewrite()
475 rewriter.replaceOp(parallelOp, loopResults); in matchAndRewrite()
517 rewriter.replaceOp(whileOp, condOp.args()); in matchAndRewrite()
566 rewriter.replaceOp(whileOp, condOp.args()); in matchAndRewrite()
/external/llvm-project/mlir/lib/Transforms/
DBufferize.cpp63 rewriter.replaceOp(op, adaptor.memref()); in matchAndRewrite()
79 rewriter.replaceOp(op, adaptor.tensor()); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dbufferize.cc56 rewriter.replaceOp(op, {buffer}); in matchAndRewrite()
71 rewriter.replaceOp(op, {buffer}); in matchAndRewrite()
/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DBufferize.cpp149 rewriter.replaceOp(genericOp, outputs); in finalizeBufferAllocationForGenericOp()
169 rewriter.replaceOp(linalgOp, outputs); in finalizeBufferAllocation()
270 rewriter.replaceOp(op, alloc); in matchAndRewrite()
313 rewriter.replaceOp(op, destMemRef); in matchAndRewrite()
DGeneralization.cpp83 rewriter.replaceOp(rootOp, genericOp.getResults()); in matchAndRewrite()
125 rewriter.replaceOp(rootOp, genericOp.getResults()); in matchAndRewrite()
/external/llvm-project/mlir/lib/Conversion/PDLToPDLInterp/
DPDLToPDLInterp.cpp85 void generateRewriter(pdl::ReplaceOp replaceOp,
541 pdl::ReplaceOp replaceOp, DenseMap<Value, Value> &rewriteValues, in generateRewriter() argument
547 if (Value replOp = replaceOp.replOperation()) in generateRewriter()
550 origOperands = replaceOp.replValues(); in generateRewriter()
554 builder.create<pdl_interp::EraseOp>(replaceOp.getLoc(), in generateRewriter()
555 mapRewriteValue(replaceOp.operation())); in generateRewriter()
563 replaceOp.getLoc(), mapRewriteValue(replaceOp.operation()), replOperands); in generateRewriter()
/external/llvm-project/mlir/lib/Dialect/SCF/Transforms/
DStructuralTypeConversions.cpp71 rewriter.replaceOp(op, newOp.getResults()); in matchAndRewrite()
114 rewriter.replaceOp(op, newOp.getResults()); in matchAndRewrite()
/external/llvm-project/mlir/lib/Conversion/GPUCommon/
DOpToFuncCallLowering.h71 rewriter.replaceOp(op, {callOp.getResult(0)}); in matchAndRewrite()
77 rewriter.replaceOp(op, {truncated}); in matchAndRewrite()
/external/llvm-project/mlir/lib/Conversion/LinalgToLLVM/
DLinalgToLLVM.cpp149 rewriter.replaceOp(op, desc); in matchAndRewrite()
192 rewriter.replaceOp(op, {desc}); in matchAndRewrite()
251 return rewriter.replaceOp(op, {desc}), success(); in matchAndRewrite()
282 rewriter.replaceOp(op, {desc}); in matchAndRewrite()
/external/llvm-project/mlir/lib/IR/
DPatternMatch.cpp161 void PatternRewriter::replaceOp(Operation *op, ValueRange newValues) { in replaceOp() function in PatternRewriter
247 return replaceOp(op, newOp->getResult(0)); in replaceOpWithResultsOfAnotherOp()
248 return replaceOp(op, newOp->getResults()); in replaceOpWithResultsOfAnotherOp()
/external/llvm-project/mlir/lib/Dialect/Vector/
DVectorTransforms.cpp726 rewriter.replaceOp(xferReadOp, newVec); in matchAndRewrite()
981 rewriter.replaceOp(shapeCastOp, producer); in matchAndRewrite()
1008 rewriter.replaceOp(shapeCastOp, sourceShapeCastOp.source()); in matchAndRewrite()
1022 rewriter.replaceOp(tupleGetOp, producer); in matchAndRewrite()
1121 rewriter.replaceOp(op, result); in matchAndRewrite()
1166 rewriter.replaceOp(op, result); in matchAndRewrite()
1181 rewriter.replaceOp(op, op.source()); in matchAndRewrite()
1226 rewriter.replaceOp(op, result); in matchAndRewrite()
1281 rewriter.replaceOp(op, expandIndices(loc, resType, 0, transp, lhs, rhs, in matchAndRewrite()
1345 rewriter.replaceOp(op, genMult(loc, op.lhs(), b, acc, isInt, rewriter)); in matchAndRewrite()
[all …]
/external/llvm-project/mlir/lib/Conversion/AVX512ToLLVM/
DConvertAVX512ToLLVM.cpp57 return rewriter.replaceOp(op, newOp->getResult(0)), success(); in matchAndRewriteOneToOne()
68 rewriter.replaceOp(op, results); in matchAndRewriteOneToOne()
/external/llvm-project/mlir/test/lib/Dialect/Tosa/
DTosaTestPasses.cpp94 rewriter.replaceOp(op, {newNegateOp.getResult()}); in matchAndRewrite()
174 rewriter.replaceOp(op, {newTosaRescaleOp.getResult()}); in matchAndRewrite()
/external/llvm-project/mlir/examples/toy/Ch3/mlir/
DToyCombine.cpp50 rewriter.replaceOp(op, {transposeInputOp.getOperand()}); in matchAndRewrite()
/external/llvm-project/mlir/lib/Dialect/StandardOps/Transforms/
DExpandOps.cpp68 rewriter.replaceOp(op, genericOp.getResult()); in matchAndRewrite()
158 rewriter.replaceOp(op, {res}); in matchAndRewrite()
207 rewriter.replaceOp(op, {res}); in matchAndRewrite()
/external/llvm-project/mlir/lib/Conversion/VectorToLLVM/
DConvertVectorToLLVM.cpp639 rewriter.replaceOp( in matchAndRewrite()
685 rewriter.replaceOp(op, shuffle); in matchAndRewrite()
704 rewriter.replaceOp(op, insert); in matchAndRewrite()
760 rewriter.replaceOp(op, extracted); in matchAndRewrite()
783 rewriter.replaceOp(op, extracted); in matchAndRewrite()
876 rewriter.replaceOp(op, inserted); in matchAndRewrite()
911 rewriter.replaceOp(op, inserted); in matchAndRewrite()
959 rewriter.replaceOp(op, desc); in matchAndRewrite()
1052 rewriter.replaceOp(op, op.source()); in matchAndRewrite()
1084 rewriter.replaceOp(op, res); in matchAndRewrite()
[all …]
/external/llvm-project/mlir/examples/toy/Ch6/mlir/
DToyCombine.cpp55 rewriter.replaceOp(op, {transposeInputOp.getOperand()}); in matchAndRewrite()

12345