- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for has_static_range_ptq (0.51 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/cc/config_test.cc
EXPECT_TRUE(default_spec.method().has_static_range_ptq()); // Test that the expansion for convolution ops is done. const QuantizationSpec& conv_spec = new_config.specs().specs(1); EXPECT_THAT(conv_spec.matcher().function_name().regex(), StrEq("composite_conv.*")); ASSERT_TRUE(conv_spec.method().has_static_range_ptq());
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/cc/report_test.cc
const QuantizationResult& result = results.results(0); EXPECT_THAT(result.quantizable_unit().name(), StrEq("composite_dot_general_fn")); EXPECT_TRUE(result.method().has_static_range_ptq()); } TEST_F(QuantizationReportTest, InitializeWithModuleOpWithoutQuantizationMethodAttribute) { // A quantized dot_general op but the `CallOp` is missing the // `_quantization_method` attribute.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 10:10:34 UTC 2024 - 18.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc
enable_per_channel_quantized_weight) {} LogicalResult match(func::FuncOp entry_func_op, const Method& quantization_method) const override { if (!quantization_method.has_static_range_ptq()) { return failure(); } return MatchGemmStyleOp<DotGeneralOp>(entry_func_op); } void rewrite(func::FuncOp entry_func_op, const Method& quantization_method,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/ops/stablehlo_op_quant_spec.cc
void PopulateCoeffOpQuantDimIfPerChannelQuantized( TF::XlaCallModuleOp xla_call_module_op, OpQuantSpec& spec) { absl::StatusOr<Method> method = GetQuantizationMethod(xla_call_module_op); if (method.ok() && method->has_static_range_ptq()) { // TODO: b/331145946 - Use `Method` accessors. const StaticRangePtq& static_range_ptq_spec = method->static_range_ptq(); // Look for quantized dimension specs for each quantized type and
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/instrumentations/save_report_test.cc
ASSERT_THAT(results.results(), SizeIs(1)); EXPECT_THAT(results.results(0).quantizable_unit().name(), StrEq("composite_dot_general_fn")); EXPECT_TRUE(results.results(0).method().has_static_range_ptq()); } TEST_F(SaveQuantizationReportInstrumentationTest, ReportNotSavedWhenNoQuantizeCompositeFunctionsPass) { constexpr absl::string_view kModuleWithCompositeDotGeneral = R"mlir(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 02:59:01 UTC 2024 - 9.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_custom_aggregation_ops.cc
if (auto xla_call_module_op = dyn_cast_or_null<TF::XlaCallModuleOp>(op); xla_call_module_op != nullptr) { absl::StatusOr<Method> method = GetQuantizationMethod(xla_call_module_op); if (method.ok() && method->has_static_range_ptq()) return true; } TF::PartitionedCallOp call_op = dyn_cast_or_null<TF::PartitionedCallOp>(op); return call_op && call_op->hasAttrOfType<StringAttr>(kQuantTraitAttrName) &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 14.8K bytes - Viewed (0)