- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for input_quantized_types (0.32 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/insert_weight_param.cc
} static bool IsPerTensor(const WeightOnlyPtq& weight_only_ptq) { const auto& input_quantized_types = weight_only_ptq.input_quantized_types(); if (input_quantized_types.empty()) { return true; } auto weight_type = input_quantized_types.find(1); if (weight_type == input_quantized_types.end()) { return true; } return weight_type->second.has_per_tensor(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 05:56:10 UTC 2024 - 10.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/quantize_composite_functions_weight_only.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 05:56:10 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/config_test.cc
const StaticRangePtq& srq_spec = conv_spec.method().static_range_ptq(); ASSERT_THAT(srq_spec.input_quantized_types(), SizeIs(1)); ASSERT_TRUE(srq_spec.input_quantized_types().contains(1)); ASSERT_TRUE(srq_spec.input_quantized_types().at(1).has_dimension_specs()); const QuantizedDimension& dimension_specs = srq_spec.input_quantized_types().at(1).dimension_specs(); ASSERT_TRUE(dimension_specs.has_dimension());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 06:59:34 UTC 2024 - 12K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/insert_weight_param.mlir
Sout = [#tf_type.shape<1x2x2x2>], _entry_function = @composite_conv_fn, _original_entry_function = "composite_conv_fn", _stablehlo_module_attrs = {}, _quantization_method = "weight_only_ptq {input_quantized_types {key: 1, value {per_tensor {}}}}", device = "", dim_args_spec = [], disabled_checks = [], has_token_input_output = false, module = "", platforms = [], version = 5 : i64
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 05:56:10 UTC 2024 - 22K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/prepare_quantize/prepare_quantize_per_channel.mlir
// Represents a per-channel quantization for the operand index 1 with // quantization dimension of 3 _quantization_method = "static_range_ptq {input_quantized_types {key: 1, value {dimension_specs {dimension: 3}}}}", platforms = [], version = 4 : i64 } : (tensor<1x3x2x3xf32>, tensor<2x3x3x2xf32>, tensor<2xf32>) -> tensor<1x2x2x2xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 26 07:48:15 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/quantize_composite_functions.mlir
// ----- // Tests that basic convolution is properly quantized. It is per-channel // quantized unless `enable-per-channel-quantized-weight=false`, according to // `_quantization_method` with an `input_quantized_types` and explicit // `dimension_specs`. module attributes {tf_saved_model.semantics} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 05:56:10 UTC 2024 - 91.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/quantization_config.proto
// here will be quantized with best effort. map<int32, QuantizedType> input_quantized_types = 1; } message WeightOnlyPtq { // Operand index -> QuantizedType mapping. Operands that are not specified // here will be quantized with best effort. map<int32, QuantizedType> input_quantized_types = 1; } // Represents a matching method that matches quantizable units by lifted
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 14.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/config.cc
// weights, on the channel dimension. // // In textproto, the spec corresponds to: // // { // {matcher {function_name {regex: "composite_conv.*"}}} // {method {static_range_ptq // {input_quantized_types { // key: 1, // value {dimension_specs {dimension: 3}}}} // }} // } QuantizationSpec GetPtqSpecForConvolution(Method::MethodCase method_case) { QuantizationSpec spec{};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 8.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/ops/stablehlo_op_quant_spec.cc
// Look for quantized dimension specs for each quantized type and // populate `coeff_op_quant_dim`. for (const auto& [operand_idx, quantized_type] : static_range_ptq_spec.input_quantized_types()) { if (quantized_type.has_dimension_specs()) { spec.coeff_op_quant_dim[operand_idx] = quantized_type.dimension_specs().dimension(); } } } } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 05:56:10 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/ops/stablehlo_op_quant_spec_test.cc
{ _entry_function = @composite_conv_fn_1, _original_entry_function = "composite_conv_fn_1", _quantization_method = "static_range_ptq {input_quantized_types {key: 1, value {dimension_specs {dimension: 3}}}}", _stablehlo_module_attrs = {jax.uses_shape_polymorphism = true}, _tfl_quant_trait = "fully_quantizable"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 04 07:19:09 UTC 2024 - 14.8K bytes - Viewed (0)