Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for MatchAnyOpTypeTag (0.29 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
Back to top