Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 174 for onStop (0.13 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/passes/unfreeze_constants.cc

    // ReadVariableOp patterns. The ConstOps are not erased. Returns the ConstOp ->
    // shared_name mapping. The shared_name is the shared name of the corresponding
    // VarHandleOp.
    llvm::MapVector<TF::ConstOp, std::string> ReplaceConstOpUsesWithVariableReads(
        llvm::ArrayRef<TF::ConstOp> target_const_ops) {
      llvm::MapVector<TF::ConstOp, std::string> const_op_name_map{};
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Aug 16 15:04:53 UTC 2023
    - 14K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_weights.cc

    class QuantizeConstWeights : public OpRewritePattern<TF::ConstOp> {
     public:
      explicit QuantizeConstWeights(
          MLIRContext* context,
          const tensorflow::quantization::QuantizationOptions& quantization_options)
          : OpRewritePattern<TF::ConstOp>(context),
            quant_options_(quantization_options) {}
    
      LogicalResult matchAndRewrite(TF::ConstOp op,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 05 07:39:40 UTC 2024
    - 11.3K bytes
    - Viewed (0)
  3. src/encoding/json/scanner.go

    	s.parseState = s.parseState[0:0]
    	s.err = nil
    	s.endTop = false
    }
    
    // eof tells the scanner that the end of input has been reached.
    // It returns a scan status just as s.step does.
    func (s *scanner) eof() int {
    	if s.err != nil {
    		return scanError
    	}
    	if s.endTop {
    		return scanEnd
    	}
    	s.step(s, ' ')
    	if s.endTop {
    		return scanEnd
    	}
    	if s.err == nil {
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Fri Sep 08 19:04:28 UTC 2023
    - 16.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/tensorflow/calibrator/calibration_statistics_saver_op_test.cc

                      .Input(inputs)
                      .Attr("ids", ids)
                      .Attr("calibration_methods", calibration_methods)
                      .Finalize(node_def()));
      ASSERT_THAT(InitOp(),
                  StatusIs(tsl::error::INVALID_ARGUMENT,
                           HasSubstr("NodeDef missing attr 'output_file_path'")));
    }
    
    TEST_F(CalibrationStatisticsSaverTest, WrongNumInputs) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 13 01:31:23 UTC 2024
    - 11.4K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc

          TransposeOp transpose = dyn_cast<TransposeOp>(user);
          if (!transpose) return;
    
          // With permutation defined by constant operation.
          ConstOp perm =
              dyn_cast_or_null<ConstOp>(transpose.getOperand(1).getDefiningOp());
          if (!perm) return;
    
          // With the same permutation indices.
          auto dense_elem_attr = mlir::dyn_cast<DenseElementsAttr>(perm.getValue());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/tensorflow/ops/tf_quantize_op.cc

          unique_func_name, original_point, result_type);
      return end_call_op;
    }
    
    QuantizedType CalculateUniformQuantParams(
        PatternRewriter& rewriter, TF::ConstOp op,
        tensorflow::quantization::QuantizationComponentSpec& weight_spec) {
      // TODO - b/278949920: Enable Per-Channel Quantization for XLA Opset
      // Currently, support symmetric, per-tensor, signed int8
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 11K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc

          return failure();
        }
    
        TF::ConstOp scale_op;
        TF::ConstOp zp_op;
    
        // Reads quantization parameters from the quantized type, and converts
        // them to constants.
        rewriter.setInsertionPoint(qparams_op);
        Location loc = qparams_op->getLoc();
        if (auto qtype = cast_qtype.dyn_cast<quant::UniformQuantizedType>()) {
          scale_op = rewriter.create<TF::ConstOp>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Nov 21 16:55:41 UTC 2023
    - 38.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc

    Value CreateZeroInput(Location loc, OpBuilder& builder, Attribute zero_attr,
                          DenseIntElementsAttr shape_attr) {
      ConstOp zero = builder.create<ConstOp>(loc, zero_attr);
      zero->setAttr(kICIWeightDistributionMlirBridgeMarker,
                    builder.getBoolAttr(true));
      ConstOp shape = builder.create<ConstOp>(loc, shape_attr);
      shape->setAttr(kICIWeightDistributionMlirBridgeMarker,
                     builder.getBoolAttr(true));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 13 18:52:07 UTC 2024
    - 13.9K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.cc

      auto out_segids_cst = builder.create<TFL::ConstOp>(
          builder.getI32TensorAttr(flattened_out_segids));
      auto contracting_segids_cst = builder.create<TFL::ConstOp>(
          builder.getI32TensorAttr(flattened_contracting_segids));
      auto num_segids_tensor =
          builder.create<TFL::ConstOp>(DenseIntElementsAttr::get(
              RankedTensorType::get({}, builder.getIntegerType(32)), 1));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.2K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types.cc

        }
        rewriter.replaceOp(op, new_results);
        return success();
      }
    };
    
    class TFConstOpQuantToIntPattern : public OpConversionPattern<TF::ConstOp> {
     public:
      using OpConversionPattern::OpConversionPattern;
    
      LogicalResult matchAndRewrite(
          TF::ConstOp op, TF::ConstOpAdaptor adaptor,
          ConversionPatternRewriter &rewriter) const override {
        if (!IsIllegalType(op.getOutput().getType())) 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