- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for WeightOnlyPtqComponent (0.32 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/cc/weight_only_ptq.cc
using ::tensorflow::quantization::PyFunctionLibrary; using ::tensorflow::quantization::RunPasses; WeightOnlyPtqComponent::WeightOnlyPtqComponent(absl::Nonnull<MLIRContext*> ctx) : ctx_(ABSL_DIE_IF_NULL(ctx)) {} // Crash OK absl::StatusOr<ModuleOp> WeightOnlyPtqComponent::Run( ModuleOp module_op, const QuantizationConfig& config) { TF_RETURN_IF_ERROR(RunPasses( kName, /*add_passes_func=*/
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 02:59:01 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/weight_only_ptq.h
// `TF::XlaCallModuleOp`s. They are quantized using the weight constants, not // relying on calibration. class WeightOnlyPtqComponent : public Component { public: // Used for debugging purposes. static constexpr absl::string_view kName = "quant_ptq_weight_only"; explicit WeightOnlyPtqComponent(absl::Nonnull<MLIRContext*> ctx); absl::StatusOr<ModuleOp> Run( ModuleOp module_op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 12:18:22 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/stablehlo/quantization.cc
#include "tensorflow/core/protobuf/meta_graph.pb.h" namespace tensorflow { namespace { using ::mlir::quant::stablehlo::StaticRangePtqComponent; using ::mlir::quant::stablehlo::WeightOnlyPtqComponent; using ::stablehlo::quantization::Method; using ::stablehlo::quantization::PopulateDefaults; using ::stablehlo::quantization::QuantizationConfig; using ::tensorflow::SignatureDef;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 10:49:12 UTC 2024 - 7.3K bytes - Viewed (0)