- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for MatchAnyOpTypeTag (0.29 sec)
-
tensorflow/compiler/mlir/lite/transforms/push_transpose_through_ewise.cc
// unchanged. class CommuteBothInputsTransposedWithEwiseOps : public RewritePattern { public: explicit CommuteBothInputsTransposedWithEwiseOps(MLIRContext *context) : RewritePattern(MatchAnyOpTypeTag(), /*benefit=*/1, context) {} LogicalResult matchAndRewrite(Operation *op, PatternRewriter &rewriter) const override { if (!IsElementwise(op)) { return failure(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types.cc
TFQuantTypeConverter &converter_; }; class TFQuantTypePattern : public ConversionPattern { public: TFQuantTypePattern(MLIRContext *ctx, TypeConverter &converter) : ConversionPattern(converter, MatchAnyOpTypeTag(), 1, ctx) {} LogicalResult matchAndRewrite( Operation *op, ArrayRef<Value> operands, ConversionPatternRewriter &rewriter) const override { // This pattern only handle non-UQ, non-const ops.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_tf.cc
// implemented in C++ don't do so this comparison doesn't work in those // cases. // 2) Descending pattern benefit. // 3) Op specific patterns over patterns with MatchAnyOpTypeTag. // 4) Order of patterns in `RewritePatternSet`. // Add TF->HLO legalization patterns. PopulateLegalizeTfPatterns(context, &legalize_lower_patterns); // Add TF->TF lowering patterns.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 10.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_custom_aggregation_ops.cc
// outlives this object. explicit AddCustomAggregationOp(MLIRContext *context, const CalibrationOptions &calib_opts) : RewritePattern(MatchAnyOpTypeTag(), /*benefit=*/1, context), calib_opts_(calib_opts) {} LogicalResult matchAndRewrite(Operation *op, PatternRewriter &rewriter) const override {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
} }; class ConvertGeneralOp : public ConversionPattern { public: ConvertGeneralOp(MLIRContext *context, const DenseSet<StringRef> &func_symbols) : ConversionPattern(MatchAnyOpTypeTag(), PatternBenefit(1), context), func_symbols_(func_symbols) {} LogicalResult matchAndRewrite( Operation *op, llvm::ArrayRef<Value> operands, ConversionPatternRewriter &rewriter) const final {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h
// the debug model creation. class RemoveDebugAttrPattern : public RewritePattern { public: explicit RemoveDebugAttrPattern(MLIRContext* context) : RewritePattern(MatchAnyOpTypeTag(), /*benefit=*/1, context) {} LogicalResult matchAndRewrite(Operation* op, PatternRewriter& rewriter) const override; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 41.7K bytes - Viewed (0)