- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 19 for src_saved_model_path_ (0.41 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/cc/calibration/component.cc
: ctx_(ABSL_DIE_IF_NULL(ctx)), // Crash OK py_function_lib_(ABSL_DIE_IF_NULL(py_function_lib)), // Crash OK src_saved_model_path_(src_saved_model_path), function_aliases_(std::move(function_aliases)), tags_(std::move(tags)), signature_def_map_(std::move(signature_def_map)), signature_keys_(std::move(signature_keys)) {} absl::Status CalibrationComponent::ExportToSavedModel(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 9.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/calibration/component.h
// from the python level using pybind11. absl::Nonnull<const tensorflow::quantization::PyFunctionLibrary*> py_function_lib_; // Path to the pre-calibrated SavedModel. std::string src_saved_model_path_; // Function alias mapping for pre-calibrated SavedModel. Used to preserve // aliased functions. absl::flat_hash_map<FunctionName, FunctionAlias> function_aliases_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/pywrap_quantize_model.cc
} py_function_library.SaveExportedModel( dst_saved_model_path, *exported_model, src_saved_model_path, tags, signature_def_map); return absl::OkStatus(); }, R"pbdoc( Quantizes a model that went through quantization-aware training (QAT) saved at `src_saved_model_path`. The resulting model will be saved to
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 09 06:33:29 UTC 2024 - 12K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/static_range_ptq.cc
function_aliases = GetFunctionAliases(src_saved_model_path, tags); if (!function_aliases.ok()) { return absl::InternalError(absl::StrCat( "Failed to get function alias: ", function_aliases.status().message())); } TF_ASSIGN_OR_RETURN( OwningOpRef<ModuleOp> module, ImportSavedModel(src_saved_model_path, signature_keys, tags,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/pywrap_quantize_model_test.py
src_saved_model_path = self.create_tempdir().full_path dst_saved_model_path = self.create_tempdir().full_path signature_def_keys = ['serving_default'] invalid_quant_opts_object = ('a', 'b', 'c') with self.assertRaisesRegex(TypeError, 'incompatible function arguments'): pywrap_quantize_model.quantize_ptq_static_range( src_saved_model_path, dst_saved_model_path,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 05 02:09:24 UTC 2024 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/python/quantization.py
return True return False # TODO: b/310594193 - Export API to pip package. def quantize_saved_model( src_saved_model_path: str, dst_saved_model_path: str, config: qc.QuantizationConfig, ) -> None: """Quantizes a saved model. Args: src_saved_model_path: Path to the directory for the source SavedModel. dst_saved_model_path: Path to the directory for the destination SavedModel.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 12:53:33 UTC 2024 - 4.6K bytes - Viewed (1) -
tensorflow/compiler/mlir/quantization/stablehlo/python/pywrap_quantization.cc
// LINT.IfChange(static_range_ptq) m.def("static_range_ptq", &PywrapQuantizeStaticRangePtq, R"pbdoc( Runs static-range post-training quantization (PTQ) on a SavedModel at `src_saved_model_path` and saves the resulting model to `dst_saved_model_path`. The user should pass a serialized `QuantizationConfig` for the `quantization_config_serialized` argument, and a signature key ->
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 13:51:40 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/weight_only_ptq.cc
function_aliases = GetFunctionAliases(src_saved_model_path, tags); if (!function_aliases.ok()) { return absl::InternalError(absl::StrCat( "Failed to get function alias: ", function_aliases.status().message())); } TF_ASSIGN_OR_RETURN( auto module, ImportSavedModel(src_saved_model_path, signature_keys, tags,
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/python/pywrap_quantization_lib.h
namespace stablehlo::quantization::pywrap { // Function used by the pywrap_quantization module to mirror // `::mlir::quant::stablehlo::QuantizeStaticRangePtq`. absl::Status PywrapQuantizeStaticRangePtq( absl::string_view src_saved_model_path, absl::string_view dst_saved_model_path, const QuantizationConfig& config, const std::vector<std::string>& signature_keys, const absl::flat_hash_map<std::string, tensorflow::SignatureDef>&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 13:51:40 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/pywrap_quantize_model.pyi
from tensorflow.compiler.mlir.quantization.tensorflow.python import representative_dataset as rd # LINT.IfChange(quantize_qat_model) def quantize_qat_model( src_saved_model_path: str, dst_saved_model_path: str, quantization_options_serialized: bytes, *, signature_keys: list[str], signature_def_map_serialized: dict[str, bytes],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 07 03:47:17 UTC 2024 - 2.5K bytes - Viewed (0)