Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 17 for quantized_dimension (0.25 sec)

  1. tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/importer_test_min_max.cc

      // FB-SAME:  21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0,
      // FB-SAME:  32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0 ],
      // FB-NEXT:        quantized_dimension: 1
      // FB-NEXT:      }
    
      // FB-LABEL:     name: "tfl.fully_connected:1",
      // FB-NEXT:      quantization: {
      // FB-EMPTY:
      // FB-NEXT:      }
    
      // FB-LABEL:      operators: [ {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 6.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc

        TypedValue<TensorType> scales_value,
        TypedValue<TensorType> zero_points_value, FloatType expressed_type,
        int64_t storage_type_min, int64_t storage_type_max,
        int64_t quantized_dimension, PatternRewriter &rewriter) {
      // Check whether the scales operand has constant op.
      DenseFPElementsAttr scales;
      if (!matchPattern(scales_value, m_Constant(&scales))) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 30.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/utils/const_tensor_utils.cc

      if (quant_params.scale.size() != 1) {
        return mlir::quant::UniformQuantizedPerAxisType::get(
            flags, storage_type, builder.getF32Type(), scales,
            quant_params.zero_point, quant_params.quantized_dimension, storage_min,
            storage_max);
      }
      return mlir::quant::UniformQuantizedType::get(
          flags, storage_type, builder.getF32Type(), scales[0],
          quant_params.zero_point.at(0), storage_min, storage_max);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 16.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/quantization/lite/quantize_weights_test.cc

            // The dequantize should still be quantized per-channel
            EXPECT_EQ(quant_tensor->quantization()->scale()->size(), 5);
            EXPECT_EQ(quant_tensor->quantization()->quantized_dimension(), 0);
          } else if (i == dequant_output_idx) {
            EXPECT_EQ(quant_tensor->type(), TensorType_FLOAT32);
          } else if (IsModelInputOrOutput(output_model, i)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 23:15:24 UTC 2024
    - 32.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/ir/tfl_ops.cc

      if (per_axis_qtype) {
        int32_t quantized_dimension = per_axis_qtype.getQuantizedDimension();
        for (int i = 0; i < output_shape.size(); ++i) {
          const APInt perm_val = perm_value_it[i];
          if (perm_val.getSExtValue() == quantized_dimension) {
            quantized_dimension = i;
            break;
          }
        }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

            tensorflow::GetTypeFromTFTensorShape(axis_stats_shape, b.getF32Type()),
            min_maxs);
        // TODO(fengliuai): this quantization dimension isn't correct.
        axis = b.getI64IntegerAttr(tensor.quantization->quantized_dimension);
      }
      return b.create<mlir::quantfork::StatisticsOp>(b.getUnknownLoc(), res,
                                                     layer_stats, axis_stats, axis);
    }
    
    // Returns true if this is a basic LSTM op.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/schema/schema_generated.h

      auto _details_type = _o->details.type;
      auto _details = _o->details.Pack(_fbb);
      auto _quantized_dimension = _o->quantized_dimension;
      return tflite::CreateQuantizationParameters(
          _fbb,
          _min,
          _max,
          _scale,
          _zero_point,
          _details_type,
          _details,
          _quantized_dimension);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 1M bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/common/uniform_quantized_types_test.cc

          quant::UniformQuantizedPerAxisType::get(
              /*flags=*/QuantizationFlags::Signed, builder_.getI8Type(),
              builder_.getF32Type(),
              /*scales=*/{1.0},
              /*zeroPoints=*/{0}, /*quantizedDimension=*/0, /*storageTypeMin=*/-128,
              /*storageTypeMax=*/127);
      EXPECT_TRUE(IsI8F32UniformQuantizedPerAxisType(qi8_per_axis_type));
      EXPECT_FALSE(IsI8F32UniformQuantizedType(qi8_per_axis_type));
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 28.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

          builder_, builder_.CreateVector<float>(mins),
          builder_.CreateVector<float>(maxs), /*scale=*/0, /*zero_point=*/0,
          tflite::QuantizationDetails_NONE, /*details=*/0,
          /*quantized_dimension=*/axis.has_value() ? axis.value() : 0);
    }
    
    std::optional<BufferOffset<tflite::SubGraph>> Translator::BuildSubGraph(
        const std::string& name, Region* region, const int index) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/common/ir/FakeQuantSupport.cc

                                                     nudgedZeroPoint, qmin, qmax);
    }
    
    quant::UniformQuantizedPerAxisType fakeQuantAttrsToType(
        Location loc, unsigned numBits, int32_t quantizedDimension,
        ArrayRef<double> rmins, ArrayRef<double> rmaxs, bool narrowRange,
        Type expressedType, bool isSigned) {
      size_t axisSize = rmins.size();
      if (axisSize != rmaxs.size()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 21 11:52:27 UTC 2024
    - 7.7K bytes
    - Viewed (0)
Back to top