Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for constant_ops (0.22 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

    class RewriteQuantizedConstantOp
        : public OpRewritePattern<stablehlo::ConstantOp> {
     public:
      using OpRewritePattern<stablehlo::ConstantOp>::OpRewritePattern;
    
      LogicalResult match(stablehlo::ConstantOp op) const override {
        return success(IsQuantizedTensorType(op.getOutput().getType()));
      }
    
      void rewrite(stablehlo::ConstantOp op,
                   PatternRewriter& rewriter) const override {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/optimize.cc

            added_value.reshape(RankedTensorType::get(
                {num_dims},
                mlir::cast<ShapedType>(added_value.getType()).getElementType()));
        ::mlir::arith::ConstantOp new_end = rewriter.create<arith::ConstantOp>(
            strided_slice_op.getEnd().getLoc(), new_added_value);
    
        if (strided_slice_op.getBeginMask() != 0) return failure();
        if (strided_slice_op.getEndMask() != 0) return failure();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 00:40:15 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

          {pad_op.getEdgePaddingLow().size()}, rewriter.getI64Type());
      auto slice_begins_const_op = rewriter.create<arith::ConstantOp>(
          loc, slice_attr_type,
          DenseIntElementsAttr::get(slice_attr_type, slice_begins));
      auto slice_sizes_const_op = rewriter.create<arith::ConstantOp>(
          loc, slice_attr_type,
          DenseIntElementsAttr::get(slice_attr_type, slice_sizes));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

    }
    
    static bool IsConst(Operation* op) {
      return isa<mlir::func::ConstantOp, mlir::arith::ConstantOp, mlir::TF::ConstOp,
                 tfl::ConstOp, tfl::QConstOp, tfl::SparseConstOp,
                 tfl::SparseQConstOp, mlir::TFL::NoValueOp,
                 mlir::stablehlo::ConstantOp, mlir::vhlo::ConstantOpV1>(op);
    }
    
    static bool IsTFResourceOp(Operation* op) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

      LogicalResult matchAndRewrite(ConstOp const_op,
                                    PatternRewriter& rewriter) const override {
        if (arith::ConstantOp::isBuildableWith(const_op.getValue(),
                                               const_op.getType())) {
          rewriter.replaceOpWithNewOp<arith::ConstantOp>(const_op,
                                                         const_op.getValue());
          return success();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
Back to top