Home
last modified time | relevance | path

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

12345

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_fused_conv_rewriter_test.cc448 GpuConvRewriter rewriter; in TEST_F() local
480 GpuConvRewriter rewriter; in TEST_F() local
519 GpuConvRewriter rewriter; in TEST_F() local
570 GpuConvRewriter rewriter; in TEST_F() local
614 GpuConvRewriter rewriter; in TEST_F() local
666 GpuConvRewriter rewriter; in TEST_F() local
704 GpuConvRewriter rewriter; in TEST_F() local
743 GpuConvRewriter rewriter; in TEST_F() local
784 GpuConvRewriter rewriter; in TEST_F() local
822 GpuConvRewriter rewriter; in TEST_F() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dtopk_rewriter_test.cc133 TopkRewriter rewriter( in TEST_F() local
164 TopkRewriter rewriter( in TEST_F() local
195 TopkRewriter rewriter( in TEST_F() local
226 TopkRewriter rewriter( in TEST_F() local
257 TopkRewriter rewriter( in TEST_F() local
Dbatchnorm_expander_test.cc79 BatchNormExpander rewriter(/*rewrite_training_op=*/true, in TEST_F() local
122 BatchNormExpander rewriter(/*rewrite_training_op=*/true, in TEST_F() local
145 BatchNormExpander rewriter(/*rewrite_training_op=*/true, in TEST_F() local
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_utils.cc32 Value buildRescale(PatternRewriter& rewriter, Operation* op, in buildRescale()
55 Value buildRescaleToInt32(PatternRewriter& rewriter, Operation* op, in buildRescaleToInt32()
68 Value buildRescaleFromInt32(PatternRewriter& rewriter, Operation* op, in buildRescaleFromInt32()
83 Value buildRescaleOpConvOutput(PatternRewriter& rewriter, Operation* op, in buildRescaleOpConvOutput()
167 Value getTosaConst8bitTable(PatternRewriter& rewriter, Operation* op, in getTosaConst8bitTable()
199 Value getTosaConst16bitTable(PatternRewriter& rewriter, Operation* op, in getTosaConst16bitTable()
241 void getTosaConst32bitTable(PatternRewriter& rewriter, Operation* op, in getTosaConst32bitTable()
308 Value getTosaConstTensorSingleF32(PatternRewriter& rewriter, Operation* op, in getTosaConstTensorSingleF32()
319 Value getTosaConstTensorSingleI32(PatternRewriter& rewriter, Operation* op, in getTosaConstTensorSingleI32()
359 PatternRewriter& rewriter, in getPaddingValuesFromPadType()
[all …]
Dlegalize_common.cc105 llvm::Optional<Value> convertPackOp(PatternRewriter& rewriter, Operation* op, in convertPackOp()
309 llvm::Optional<SmallVector<Value>> convertUnpackOp(PatternRewriter& rewriter, in convertUnpackOp()
405 llvm::Optional<Value> convertSelectOp(PatternRewriter& rewriter, Operation* op, in convertSelectOp()
449 llvm::Optional<Value> convertZerosLikeOp(PatternRewriter& rewriter, in convertZerosLikeOp()
477 llvm::Optional<Value> convertMultiplyOp(PatternRewriter& rewriter, in convertMultiplyOp()
541 llvm::Optional<Value> convertSquaredDifferenceOp(PatternRewriter& rewriter, in convertSquaredDifferenceOp()
568 llvm::Optional<Value> convertRoundOp(PatternRewriter& rewriter, Operation* op, in convertRoundOp()
593 llvm::Optional<Value> convertConcatV2Op(PatternRewriter& rewriter, in convertConcatV2Op()
669 llvm::Optional<Value> convertSpaceToBatchNDOp(PatternRewriter& rewriter, in convertSpaceToBatchNDOp()
939 llvm::Optional<Value> convertBatchToSpaceNDOp(PatternRewriter& rewriter, in convertBatchToSpaceNDOp()
[all …]
Dlegalize_utils.h137 TosaOp CreateOpAndInfer(PatternRewriter& rewriter, Location loc, Type result_ty, in CreateOpAndInfer()
179 void CreateReplaceOpAndInfer(PatternRewriter& rewriter, Operation* op, in CreateReplaceOpAndInfer()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dfallback_converter.cc46 mlir::ConversionPatternRewriter &rewriter) { in ConvertCoreRTTensorHandleToFallbackTensor()
82 mlir::ConversionPatternRewriter &rewriter) { in ConvertFallbackTensorToCoreRTTensorHandle()
112 mlir::ConversionPatternRewriter &rewriter) { in ConvertCoreRTOperands()
133 mlir::ConversionPatternRewriter &rewriter) { in ConvertFallbackOperands()
Dcorert_converter.cc162 ConversionPatternRewriter *rewriter) { in ConvertOpHandler()
180 mlir::Operation *op, mlir::ConversionPatternRewriter *rewriter) { in GetDistributedContext()
197 mlir::Operation *op, mlir::ConversionPatternRewriter *rewriter) { in GetRemoteChainManager()
218 mlir::Operation *op, mlir::ConversionPatternRewriter *rewriter) { in GetLocalSideEffectChain()
255 mlir::ConversionPatternRewriter *rewriter) { in GetTaskHandle()
275 mlir::ConversionPatternRewriter *rewriter) { in GetRemoteSideEffectChain()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_quantize_helper.h140 PatternRewriter& rewriter) const override { in matchAndRewrite()
173 PatternRewriter& rewriter) const { in processRestrictScale()
238 PatternRewriter& rewriter) const { in processInputs()
282 PatternRewriter& rewriter) const { in processConstantOp()
337 PatternRewriter& rewriter) const { in replaceStatsOp()
411 PatternRewriter& rewriter) const override { in matchAndRewrite()
562 PatternRewriter& rewriter) const override { in matchAndRewrite()
Dprepare_quantize_dynamic_range.cc164 void quantizeOpAsFloat16(PatternRewriter& rewriter, arith::ConstantOp op, in quantizeOpAsFloat16()
197 bool quantizeOpAsInt8(PatternRewriter& rewriter, arith::ConstantOp op, in quantizeOpAsInt8()
252 bool insertQDQ(PatternRewriter& rewriter, arith::ConstantOp op, in insertQDQ()
307 bool quantizeOps(PatternRewriter& rewriter, arith::ConstantOp op, in quantizeOps()
327 PatternRewriter& rewriter, QuantizationUnits& quantizable_ops) const { in setAsymmetricQuantizeInputAttr()
350 bool convertToFloat16Constant(PatternRewriter& rewriter, in convertToFloat16Constant()
/external/google-smali/dexlib2/src/main/java/com/android/tools/smali/dexlib2/rewriter/
DRewriterUtils.java46 public static <T> T rewriteNullable(@Nonnull Rewriter<T> rewriter, @Nullable T value) { in rewriteNullable()
50 public static <T> Set<T> rewriteSet(@Nonnull final Rewriter<T> rewriter, in rewriteSet()
/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/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
Dchlo_legalize_to_hlo.cc82 ConversionPatternRewriter &rewriter, Location loc, Value x, in materializeChebyshevPolynomialApproximation()
102 Value materializeBesselI1eApproximation(ConversionPatternRewriter &rewriter, in materializeBesselI1eApproximation()
134 Value materializeBesselI1eApproximationF32(ConversionPatternRewriter &rewriter, in materializeBesselI1eApproximationF32()
160 Value materializeBesselI1eApproximationF64(ConversionPatternRewriter &rewriter, in materializeBesselI1eApproximationF64()
202 Value materializeWithUpcast(ConversionPatternRewriter &rewriter, Location loc, in materializeWithUpcast()
260 Value materializePolynomialApproximation(ConversionPatternRewriter &rewriter, in materializePolynomialApproximation()
278 ConversionPatternRewriter &rewriter, Location loc, ValueRange args) { in materializeErfcApproximationF64ForMagnituteGeOne()
361 ConversionPatternRewriter &rewriter, Location loc, ValueRange args) { in materializeErfApproximationF64ForMagnituteLeOne()
386 Value materializeErfApproximationF64(ConversionPatternRewriter &rewriter, in materializeErfApproximationF64()
413 Value materializeErfcApproximationF64(ConversionPatternRewriter &rewriter, in materializeErfcApproximationF64()
[all …]
Dunfuse_batch_norm.cc41 PatternRewriter& rewriter) { // NOLINT in broadcastToFeatureDim()
55 PatternRewriter &rewriter) { // NOLINT in getShapeValue()
65 PatternRewriter &rewriter) { // NOLINT in materializeEpsilon()
169 PatternRewriter& rewriter) { in createReduce()
202 PatternRewriter &rewriter) { in calculateReduceSize()
Dbroadcast_propagation.cc232 PatternRewriter &rewriter, Block *block, ValueRange values) { in setInsertionPointToEarliestPointWithAllValuesAvailable()
251 Block *parentBlock, PatternRewriter &rewriter) { in realizeBroadcastIntents()
316 PatternRewriter &rewriter) { in transitivelyEraseUnusedSideEffectFreeOps()
349 PatternRewriter &rewriter) { in propagateBroadcast()
/external/google-smali/dexlib2/src/test/java/com/android/tools/smali/dexlib2/rewriter/
DRewriteArrayTypeTest.java75 DexRewriter rewriter = new DexRewriter(new RewriterModule() { in testRewriteArrayTypeTest() local
111 DexRewriter rewriter = new DexRewriter(new RewriterModule()); in testUnmodifiedArrayTypeTest() local
/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()
DObjectsRequireNonNullMethodRewriter.java33 private final CoreLibraryRewriter rewriter; field in ObjectsRequireNonNullMethodRewriter
35 public ObjectsRequireNonNullMethodRewriter(ClassVisitor cv, CoreLibraryRewriter rewriter) { in ObjectsRequireNonNullMethodRewriter()
DLongCompareMethodRewriter.java30 private final CoreLibraryRewriter rewriter; field in LongCompareMethodRewriter
32 public LongCompareMethodRewriter(ClassVisitor cv, CoreLibraryRewriter rewriter) { in LongCompareMethodRewriter()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Deinsum.cc61 PatternRewriter* rewriter) { in createI32ConstOp()
68 Location loc, PatternRewriter* rewriter) { in createI32ConstantOp()
77 Location loc, PatternRewriter* rewriter) { in createI64ConstantOp()
86 PatternRewriter* rewriter) { in createTransposeOp()
102 PatternRewriter* rewriter) { in createReshapeOp()
117 PatternRewriter* rewriter) { in createReshapeOpForDynamic()
155 EinsumDimensionNumbers& dnums, Location loc, PatternRewriter* rewriter) { in createOutputReshapeOpForDynamic()
435 std::vector<int32_t>* out_inverse_transpose, PatternRewriter* rewriter) { in transposeForBatchMatmul()
504 PatternRewriter* rewriter) { in reshapeForBatchMatmul()
630 PatternRewriter& rewriter) { in rewriteToBatchMatmul()
Dconvert_tf_control_flow_to_scf.cc32 PatternRewriter& rewriter) { in moveBlock()
61 PatternRewriter& rewriter) { in matchAndRewrite()
130 TypeRange scf_block_arguments_type, PatternRewriter& rewriter) { in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Draise_to_tf.cc136 Value CastToNonDerivedType(PatternRewriter& rewriter, Location loc, in CastToNonDerivedType()
163 void CastValuesToSameType(PatternRewriter& rewriter, Location loc, in CastValuesToSameType()
186 PatternRewriter& rewriter, Type input_tfr_type, in AddDerivedAttrs()
229 PatternRewriter& rewriter, TFRFuncOp signature, CallOp call_op, in CollectInputsAndAttributes()
375 PatternRewriter& rewriter, CallOp call_op, in CreateAndReplaceOp()
/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
Dprepare_quantize_drq.cc145 bool quantizeOpAsInt8(PatternRewriter& rewriter, arith::ConstantOp op, in quantizeOpAsInt8()
165 bool insertQDQ(PatternRewriter& rewriter, arith::ConstantOp op, in insertQDQ()
198 bool quantizeOps(PatternRewriter& rewriter, arith::ConstantOp op, in quantizeOps()
/external/bcc/src/cc/frontends/clang/
Dtp_frontend_action.cc50 TracepointTypeVisitor::TracepointTypeVisitor(ASTContext &C, Rewriter &rewriter) in TracepointTypeVisitor()
128 TracepointTypeConsumer::TracepointTypeConsumer(ASTContext &C, Rewriter &rewriter) in TracepointTypeConsumer()
/external/angle/src/compiler/translator/tree_ops/msl/
DTransposeRowMajorMatrices.cpp45 Rewriter rewriter(compiler); in TransposeRowMajorMatricies() local

12345