Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 82 for matchAndRewrite (0.6 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc

        op->erase();
      });
    }
    
    class ConvertGraphOp : public OpConversionPattern<tfg::GraphOp> {
     public:
      using OpConversionPattern::OpConversionPattern;
    
      LogicalResult matchAndRewrite(
          tfg::GraphOp graph, OpAdaptor adaptor,
          ConversionPatternRewriter &rewriter) const final {
        Location loc = graph.getLoc();
        // To keep the import-as-graph logic taken by TFG, we create `void func()`
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/decompose_optionals.cc

      void runOnOperation() override;
    };
    
    class HandleOptionalFrom : public OpRewritePattern<TF::OptionalFromValueOp> {
     public:
      using OpRewritePattern::OpRewritePattern;
      LogicalResult matchAndRewrite(TF::OptionalFromValueOp op,
                                    PatternRewriter& rewriter) const override {
        Value value = nullptr;
        for (auto v : op.getComponents()) {
          value = v;
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 9.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/stablehlo/passes/prepare_quantize.cc

     public:
      explicit MergeConsecutiveQuantizeCast(MLIRContext* context)
          : OpRewritePattern<quantfork::QuantizeCastOp>(context) {}
    
     private:
      LogicalResult matchAndRewrite(quantfork::QuantizeCastOp q_op,
                                    PatternRewriter& rewriter) const override {
        auto preceding_qcast =
            q_op.getArg().getDefiningOp<quantfork::QuantizeCastOp>();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 03 05:11:03 UTC 2024
    - 8.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/stablehlo/passes/post_quantize.cc

        : public OpRewritePattern<quantfork::DequantizeCastOp> {
     public:
      explicit RemoveVolatileQdqPattern(MLIRContext* context)
          : OpRewritePattern<quantfork::DequantizeCastOp>(context) {}
    
      LogicalResult matchAndRewrite(quantfork::DequantizeCastOp op,
                                    PatternRewriter& rewriter) const override {
        auto input_op = op.getArg().getDefiningOp();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 05 07:39:40 UTC 2024
    - 6.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc

    }
    
    LogicalResult ConvertTFBatchMatMulOp::matchAndRewrite(
        Operation* op, PatternRewriter& rewriter) const {
      if (ConvertTFBatchMatMulOp2TFLFullyConnectedOp<TF::BatchMatMulOp>(op,
                                                                        rewriter))
        return success();
      return failure();
    }
    
    LogicalResult ConvertTFBatchMatMulV2Op::matchAndRewrite(
        Operation* op, PatternRewriter& rewriter) const {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 20:06:54 UTC 2024
    - 45.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/transforms/legalize_hashtables.cc

    class LegalizeHashTableOpPattern : public OpRewritePattern<TF::HashTableV2Op> {
     public:
      using OpRewritePattern<TF::HashTableV2Op>::OpRewritePattern;
    
      LogicalResult matchAndRewrite(TF::HashTableV2Op hashtable_op,
                                    PatternRewriter& rewriter) const override {
        auto output_type = RankedTensorType::get(
            {1}, TF::ResourceType::get(rewriter.getContext()));
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc

    // on the tensor proto.
    class ConvertUniformQuantizedDotHybridOp
        : public OpConversionPattern<TF::UniformQuantizedDotHybridOp> {
     public:
      using OpConversionPattern::OpConversionPattern;
    
      LogicalResult matchAndRewrite(
          TF::UniformQuantizedDotHybridOp op,
          TF::UniformQuantizedDotHybridOpAdaptor adaptor,
          ConversionPatternRewriter &rewriter) const override {
        // Uniform Quantized type for the rhs.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 30.9K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform_patterns.cc

    LogicalResult EnsureBiasForConv2d::matchAndRewrite(
        TFL::Conv2DOp conv_op, PatternRewriter& rewriter) const {
      return EnsureBias(conv_op, 2, rewriter);
    }
    
    // ================== slice ============================
    
    // If a slice op has < 4d dimension, will pad it to 4d.
    LogicalResult PadSlice::matchAndRewrite(TFL::SliceOp slice_op,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 25.4K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/transforms/reduce_type_precision.cc

    // the input is sign-extended two's complement.
    class CheckRangeAndConvertI8ToI4 : public OpRewritePattern<arith::ConstantOp> {
     public:
      using OpRewritePattern<arith::ConstantOp>::OpRewritePattern;
    
      LogicalResult matchAndRewrite(arith::ConstantOp op,
                                    PatternRewriter &rewriter) const override {
        auto const_type = mlir::dyn_cast<ShapedType>(op.getType());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.4K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/lite/quantization/ir/ConvertConst.cc

      LogicalResult matchAndRewrite(QuantizeCastOp qbarrier,
                                    PatternRewriter &rewriter) const override;
    };
    
    }  // namespace
    
    /// Matches a [constant] -> [qbarrier] where the qbarrier results type is
    /// quantized and the operand type is quantizable.
    
    LogicalResult QuantizedConstRewrite::matchAndRewrite(
        QuantizeCastOp qbarrier, PatternRewriter &rewriter) const {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 4.4K bytes
    - Viewed (0)
Back to top