Home
last modified time | relevance | path

Searched defs:rewriter (Results 1 – 25 of 100) sorted by relevance

1234

/external/llvm-project/mlir/include/mlir/Conversion/StandardToLLVM/
DConvertStandardToLLVM.h576 ConversionPatternRewriter &rewriter) const final { in rewrite()
584 ConversionPatternRewriter &rewriter) const final { in matchAndRewrite()
591 ConversionPatternRewriter &rewriter) const { in rewrite()
599 ConversionPatternRewriter &rewriter) const { in matchAndRewrite()
643 ConversionPatternRewriter &rewriter) const override { in matchAndRewrite()
660 ConversionPatternRewriter &rewriter) const override { in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_utils.cc29 Value buildRescale(PatternRewriter& rewriter, Operation* op, in buildRescale()
50 Value buildRescaleToInt32(PatternRewriter& rewriter, Operation* op, in buildRescaleToInt32()
64 Value buildRescaleFromInt32(PatternRewriter& rewriter, Operation* op, in buildRescaleFromInt32()
79 Value buildRescaleOpConvOutput(PatternRewriter& rewriter, Operation* op, in buildRescaleOpConvOutput()
165 Value getTosa1DConstTensorTable(PatternRewriter& rewriter, Operation* op, in getTosa1DConstTensorTable()
191 Value getTosaConstTensorSingleF32(PatternRewriter& rewriter, Operation* op, in getTosaConstTensorSingleF32()
202 Value getTosaConstTensorSingleI32(PatternRewriter& rewriter, Operation* op, in getTosaConstTensorSingleI32()
239 PatternRewriter& rewriter, ArrayAttr& explicit_padding) { in getPaddingValuesFromPadType()
285 PatternRewriter& rewriter) { in getPaddingValuesFromExplicitPadAttr()
307 ArrayAttr strides, ArrayAttr dilations, PatternRewriter& rewriter, in getTransposeConv2dPaddingValues()
[all …]
Dlegalize_common.cc43 llvm::Optional<Value> convertPackOp(PatternRewriter& rewriter, Operation* op, in convertPackOp()
261 llvm::Optional<ValueRange> convertUnpackOp(PatternRewriter& rewriter, in convertUnpackOp()
364 llvm::Optional<Value> convertSelectOp(PatternRewriter& rewriter, Operation* op, in convertSelectOp()
411 llvm::Optional<Value> convertZerosLikeOp(PatternRewriter& rewriter, in convertZerosLikeOp()
440 llvm::Optional<Value> convertMultiplyOp(PatternRewriter& rewriter, in convertMultiplyOp()
502 llvm::Optional<Value> convertSquaredDifferenceOp(PatternRewriter& rewriter, in convertSquaredDifferenceOp()
528 llvm::Optional<Value> convertRoundOp(PatternRewriter& rewriter, Operation* op, in convertRoundOp()
553 llvm::Optional<Value> convertConcatV2Op(PatternRewriter& rewriter, in convertConcatV2Op()
731 llvm::Optional<Value> convertSpaceToBatchNDOp(PatternRewriter& rewriter, in convertSpaceToBatchNDOp()
999 llvm::Optional<Value> convertBatchToSpaceNDOp(PatternRewriter& rewriter, in convertBatchToSpaceNDOp()
[all …]
/external/llvm-project/mlir/test/lib/Rewrite/
DTestPDLByteCode.cpp18 PatternRewriter &rewriter) { in customSingleEntityConstraint()
24 PatternRewriter &rewriter) { in customMultiEntityConstraint()
30 PatternRewriter &rewriter) { in customCreate()
38 PatternRewriter &rewriter) { in customRewriter()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtopk_rewriter_test.cc80 TopkRewriter rewriter([](const HloSortInstruction*, int64) { return true; }); in TEST_F() local
110 TopkRewriter rewriter([](const HloSortInstruction*, int64) { return true; }); in TEST_F() local
140 TopkRewriter rewriter([](const HloSortInstruction*, int64) { return true; }); in TEST_F() local
Dbatchnorm_expander_test.cc80 BatchNormExpander rewriter(/*rewrite_training_op=*/true, in TEST_F() local
123 BatchNormExpander rewriter(/*rewrite_training_op=*/true, in TEST_F() local
146 BatchNormExpander rewriter(/*rewrite_training_op=*/true, in TEST_F() local
/external/smali/dexlib2/src/main/java/org/jf/dexlib2/rewriter/
DRewriterUtils.java43 public static <T> T rewriteNullable(@Nonnull Rewriter<T> rewriter, @Nullable T value) { in rewriteNullable()
47 public static <T> Set<T> rewriteSet(@Nonnull final Rewriter<T> rewriter, in rewriteSet()
/external/desugar/java/com/google/devtools/build/android/desugar/
DClassReaderFactory.java26 private final CoreLibraryRewriter rewriter; field in ClassReaderFactory
28 public ClassReaderFactory(IndexedInputs indexedInputs, CoreLibraryRewriter rewriter) { in ClassReaderFactory()
DLongCompareMethodRewriter.java30 private final CoreLibraryRewriter rewriter; field in LongCompareMethodRewriter
32 public LongCompareMethodRewriter(ClassVisitor cv, CoreLibraryRewriter rewriter) { in LongCompareMethodRewriter()
DObjectsRequireNonNullMethodRewriter.java33 private final CoreLibraryRewriter rewriter; field in ObjectsRequireNonNullMethodRewriter
35 public ObjectsRequireNonNullMethodRewriter(ClassVisitor cv, CoreLibraryRewriter rewriter) { in ObjectsRequireNonNullMethodRewriter()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dchlo_legalize_to_hlo.cc82 Value MaterializePolynomialApproximation(ConversionPatternRewriter &rewriter, in MaterializePolynomialApproximation()
100 ConversionPatternRewriter &rewriter, Location loc, ValueRange args) { in MaterializeErfcApproximationF64ForMagnituteGEOne()
184 ConversionPatternRewriter &rewriter, Location loc, ValueRange args) { in MaterializeErfApproximationF64ForMagnituteLEOne()
209 Value MaterializeErfApproximationF64(ConversionPatternRewriter &rewriter, in MaterializeErfApproximationF64()
236 Value MaterializeErfcApproximationF64(ConversionPatternRewriter &rewriter, in MaterializeErfcApproximationF64()
269 ConversionPatternRewriter &rewriter, Location loc, ValueRange args) { in MaterializeErfcApproximationF32ForMagnitudeGEOne()
334 ConversionPatternRewriter &rewriter, Location loc, ValueRange args) { in MaterializeErfApproximationF32ForMagnitudeLEOne()
353 Value MaterializeErfApproximationF32(ConversionPatternRewriter &rewriter, in MaterializeErfApproximationF32()
384 Value MaterializeErfcApproximationF32(ConversionPatternRewriter &rewriter, in MaterializeErfcApproximationF32()
411 Value MaterializeWithUpcast(ConversionPatternRewriter &rewriter, Location loc, in MaterializeWithUpcast()
[all …]
Dunfuse_batch_norm.cc39 PatternRewriter& rewriter) { // NOLINT in BroadcastToFeatureDim()
55 PatternRewriter& rewriter) { // NOLINT in CalculateShapeValue()
69 PatternRewriter& rewriter) { // NOLINT in MaterializeEpsilon()
Dlower_general_dot.cc53 PatternRewriter *rewriter) { in TransposeReshape()
98 PatternRewriter *rewriter) { in ProcessDotArg()
/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DSparsification.cpp433 static Type genIntType(PatternRewriter &rewriter, linalg::SparseIntType tp) { in genIntType()
448 PatternRewriter &rewriter, linalg::GenericOp op) { in genBuffers()
513 PatternRewriter &rewriter, linalg::GenericOp op, in genTensorLoad()
542 PatternRewriter &rewriter, linalg::GenericOp op, in genTensorStore()
556 static Value genLoad(PatternRewriter &rewriter, Location loc, Value ptr, in genLoad()
566 PatternRewriter &rewriter, unsigned exp) { in genInvariantValue()
571 static Value genExp(Merger &merger, CodeGen &codegen, PatternRewriter &rewriter, in genExp()
596 PatternRewriter &rewriter, linalg::GenericOp op, in genInvariants()
624 static bool genInit(Merger &merger, CodeGen &codegen, PatternRewriter &rewriter, in genInit()
664 PatternRewriter &rewriter, linalg::GenericOp op, in genFor()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Draise_to_tf.cc125 Value CastToNonDerivedType(PatternRewriter& rewriter, Location loc, in CastToNonDerivedType()
160 void CastValuesToSameType(PatternRewriter& rewriter, Location loc, in CastValuesToSameType()
183 PatternRewriter& rewriter, Type input_tfr_type, in AddDerivedAttrs()
226 PatternRewriter& rewriter, TFRFuncOp signature, CallOp call_op, in CollectInputsAndAttributes()
359 PatternRewriter& rewriter, CallOp call_op, in CreateAndReplaceOp()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dunroll_batch_matmul.cc66 PatternRewriter& rewriter) { in createReshapeOp()
80 Value value, int batch_size, Location loc, PatternRewriter& rewriter) { in sliceInput()
124 Value value, Location loc, PatternRewriter& rewriter) { in createTransposeOp()
158 Location loc, PatternRewriter& rewriter) { in createMatMulOps()
Deinsum.cc54 PatternRewriter* rewriter) { in createTransposeOp()
73 PatternRewriter* rewriter) { in createReshapeOp()
188 std::vector<int32_t>* out_inverse_transpose, PatternRewriter* rewriter) { in transposeForBatchMatmul()
240 PatternRewriter* rewriter) { in reshapeForBatchMatmul()
308 PatternRewriter& rewriter) { in rewriteToBatchMatmul()
Dlegalize_hlo.cc449 Value ConvertDot(PatternRewriter &rewriter, Value lhs, Value rhs, in ConvertDot()
542 Value ConvertDotOp(PatternRewriter &rewriter, Operation *old_op) { in ConvertDotOp()
559 Value ConvertDotGeneralOp(PatternRewriter &rewriter, Operation *old_op) { in ConvertDotGeneralOp()
960 ConstantOp ShapeToConst(PatternRewriter &rewriter, Value value) { in ShapeToConst()
974 ConversionPatternRewriter &rewriter) { in NormalizeIndexVector()
1170 Value ConvertPadOp(PatternRewriter &rewriter, Operation *old_op) { in ConvertPadOp()
1203 ConstantOp ExpandedShape(PatternRewriter &rewriter, Value input, in ExpandedShape()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Ddilated_conv.h83 Conv2dOpTy op, PatternRewriter& rewriter) const { in matchAndRewrite()
311 PatternRewriter& rewriter) const { in ExtractDilationsAttrFromBlockShape()
/external/llvm-project/mlir/lib/Conversion/VectorToLLVM/
DConvertVectorToLLVM.cpp36 static Value insertOne(ConversionPatternRewriter &rewriter, in insertOne()
53 static Value insertOne(PatternRewriter &rewriter, Location loc, Value from, in insertOne()
64 static Value extractOne(ConversionPatternRewriter &rewriter, in extractOne()
80 static Value extractOne(PatternRewriter &rewriter, Location loc, Value vector, in extractOne()
112 static Value buildVectorComparison(ConversionPatternRewriter &rewriter, in buildVectorComparison()
163 static LogicalResult getBase(ConversionPatternRewriter &rewriter, Location loc, in getBase()
180 static LogicalResult getBasePtr(ConversionPatternRewriter &rewriter, in getBasePtr()
192 static LogicalResult getBasePtr(ConversionPatternRewriter &rewriter, in getBasePtr()
206 static LogicalResult getIndexedPtrs(ConversionPatternRewriter &rewriter, in getIndexedPtrs()
220 replaceTransferOpWithLoadOrStore(ConversionPatternRewriter &rewriter, in replaceTransferOpWithLoadOrStore()
[all …]
/external/angle/src/compiler/translator/TranslatorMetalDirect/
DTransposeRowMajorMatrices.cpp46 Rewriter rewriter(compiler); in TransposeRowMajorMatricies() local
DMapSymbols.cpp40 Rewriter rewriter(compiler, std::move(map)); in MapSymbols() local
/external/llvm-project/mlir/lib/Conversion/VectorToROCDL/
DVectorToROCDL.cpp31 ConversionPatternRewriter &rewriter, ArrayRef<Value> operands, in replaceTransferOpWithMubuf()
41 ConversionPatternRewriter &rewriter, ArrayRef<Value> operands, in replaceTransferOpWithMubuf()
/external/bcc/src/cc/frontends/clang/
Dtp_frontend_action.cc48 TracepointTypeVisitor::TracepointTypeVisitor(ASTContext &C, Rewriter &rewriter) in TracepointTypeVisitor()
229 TracepointTypeConsumer::TracepointTypeConsumer(ASTContext &C, Rewriter &rewriter) in TracepointTypeConsumer()
/external/llvm-project/mlir/lib/Transforms/Utils/
DDialectConversion.cpp291 PatternRewriter &rewriter; member
1033 Location loc, PatternRewriter &rewriter, TypeConverter *converter, in remapValues()
1572 ConversionPatternRewriter &rewriter) { in legalize()
1651 ConversionPatternRewriter &rewriter) { in legalizeWithFold()
1690 ConversionPatternRewriter &rewriter) { in legalizeWithPattern()
1722 ConversionPatternRewriter &rewriter) { in canApplyPattern()
1745 ConversionPatternRewriter &rewriter, in legalizePatternResult()
1784 Operation *op, ConversionPatternRewriter &rewriter, in legalizePatternBlockActions()
1836 ConversionPatternRewriter &rewriter, ConversionPatternRewriterImpl &impl, in legalizePatternCreatedOperations()
1850 ConversionPatternRewriter &rewriter, ConversionPatternRewriterImpl &impl, in legalizePatternRootUpdates()
[all …]

1234