Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for UniformQuantizedConvolutionOp (0.39 sec)

  1. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc

          RankedTensorType::get({static_cast<int64_t>(array_attr.size())},
                                rewriter.getIntegerType(64)),
          array);
    }
    
    template <typename UniformQuantizedConvolutionOp>
    FailureOr<ElementsAttr> ConvertPaddingAttr(
        UniformQuantizedConvolutionOp op,
        const xla::ConvolutionDimensionNumbers &dnums, PatternRewriter &rewriter) {
      StringAttr conv_padding = op.getPaddingAttr();
      SmallVector<int64_t> padding_nums;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 30.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/ops/uniform_op_quant_spec.cc

    namespace mlir::quant {
    
    std::unique_ptr<OpQuantSpec> GetUniformOpQuantSpec(Operation* op) {
      auto spec = std::make_unique<OpQuantSpec>();
      if (isa<TF::UniformQuantizedConvolutionHybridOp>(op) ||
          isa<TF::UniformQuantizedConvolutionOp>(op)) {
        spec->coeff_op_quant_dim[1] = 3;
      } else if (isa<TF::UniformQuantizedDotHybridOp>(op)) {
        spec->coeff_op_quant_dim[1] = -1;
      }
    
      for (auto quantizable_operand : spec->coeff_op_quant_dim) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun Mar 24 07:44:40 UTC 2024
    - 1.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/stablehlo/utils/tf_type_utils.cc

          TF::UniformDequantizeOp,
          TF::UniformQuantizeOp,
          TF::UniformQuantizedAddOp,
          TF::UniformQuantizedClipByValueOp,
          TF::UniformQuantizedConvolutionHybridOp,
          TF::UniformQuantizedConvolutionOp,
          TF::UniformQuantizedDotHybridOp,
          TF::UniformQuantizedDotOp,
          TF::UniformRequantizeOp
          // go/keep-sorted end
          // clang-format on
          >(op);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 4.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc

                  enable_per_channel_quantization)))
            return failure();
        } else if (auto uniform_op =
                       llvm::dyn_cast<TF::UniformQuantizedConvolutionOp>(inner_op);
                   uniform_op != nullptr) {
          if (failed(FillAttributesForUniformQuantizedConvolutionOp(
                  rewriter, uniform_op, identifier_to_attr, quantization_method,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 54.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

    }
    
    //===----------------------------------------------------------------------===//
    // UniformQuantizedConvolutionOp
    //===----------------------------------------------------------------------===//
    //
    
    LogicalResult UniformQuantizedConvolutionOp::verify() {
      return VerifyLhsRhsBothUniformQuantizedOp(*this);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
Back to top