Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 14 of 14 for MatchAnyOpTypeTag (1.01 sec)

  1. 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)
  2. tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/outline_composites.cc

    // identifying `erf` ops and validate the structure around them.
    class OutlineGELU : public RewritePattern {
     public:
      explicit OutlineGELU(MLIRContext* context)
          : RewritePattern(MatchAnyOpTypeTag(), /*benefit=*/1, context) {}
    
      LogicalResult matchAndRewrite(Operation* op,
                                    PatternRewriter& rewriter) const override {
        if (!MatchERF(op)) return failure();
        // `add 1`
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 9.6K bytes
    - Viewed (0)
  3. 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)
  4. 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