Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for ConversionPatternRewriter (0.23 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/util.cc

                          ConversionPatternRewriter& rewriter, int64_t const_value,
                          Type type) {
      Value result_const =
          builder.create<TF::ConstOp>(rewriter.getIntegerAttr(type, const_value));
      return result_const;
    }
    
    Value BuildIntArrayConstOp(ImplicitLocOpBuilder& builder,
                               ConversionPatternRewriter& rewriter,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc

    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()`
        // to contain the ops in the tfg::GraphOp. That means the arguments/results
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc

     public:
      using OpConversionPattern::OpConversionPattern;
    
      LogicalResult matchAndRewrite(
          TF::UniformQuantizedDotHybridOp op,
          TF::UniformQuantizedDotHybridOpAdaptor adaptor,
          ConversionPatternRewriter &rewriter) const override {
        // Uniform Quantized type for the rhs.
        int64_t rhs_quantized_dimension = op.getRhsQuantizationAxis();
        // Currently for dot, PTQ supports per-tensor quantization.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 30.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/scatter.h

        ShapedType& updates_type, ConversionPatternRewriter& rewriter);
    
    template <typename BinaryOp, typename TfOp>
    class ConvertScatterOp : public OpConversionPattern<mhlo::ScatterOp> {
     public:
      using OpConversionPattern::OpConversionPattern;
    
      LogicalResult matchAndRewrite(
          mhlo::ScatterOp scatter_op, OpAdaptor adaptor,
          ConversionPatternRewriter& rewriter) const final {
        {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.1K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types.cc

          : 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.
        if (IsTFUniformQuantizedOp(op) || llvm::isa<TF::ConstOp>(op)) {
          return failure();
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 12.6K bytes
    - Viewed (0)
Back to top