- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 24 for SignatureDef (0.17 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/python/quantization.py
# Mapping of signature def key -> SignatureDef. _SignatureDefMap = Mapping[str, meta_graph_pb2.SignatureDef] def _serialize_signature_def_map( signature_def_map: _SignatureDefMap, ) -> dict[str, bytes]: """Serializes SignatureDef values in `signature_def_map`. Args: signature_def_map: Signature key -> SignatureDef mapping. Returns:
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/cc/saved_model/loader.h
/// SavedModel. virtual Session* GetSession() const = 0; /// Returns a map from signature name to SignatureDef for all signatures in /// in the SavedModel. virtual const protobuf::Map<string, SignatureDef>& GetSignatures() const = 0; }; /// SavedModel representation once the SavedModel is loaded from storage. ///
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 05 18:28:37 UTC 2023 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/stablehlo/quantization.cc
using ::stablehlo::quantization::QuantizationConfig; using ::tensorflow::SignatureDef; using ::tensorflow::quantization::PyFunctionLibrary; // Returns signature key -> `SignatureDef` mapping, excluding the signature for // initialization op, which is only used during initialization. // TODO: b/314124142 - Remove the need for this function. absl::flat_hash_map<std::string, SignatureDef> GetSignatureDefMapFromBundle(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 10:49:12 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/static_range_ptq.cc
#include "tensorflow/core/protobuf/saver.pb.h" #include "tsl/platform/statusor.h" namespace mlir::quant::stablehlo { using ::stablehlo::quantization::QuantizationConfig; using ::tensorflow::SignatureDef; using ::tensorflow::quantization::ExportedModel; using ::tensorflow::quantization::PyFunctionLibrary; StaticRangePtqComponent::StaticRangePtqComponent( absl::Nonnull<MLIRContext*> ctx,
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/stablehlo/python/pywrap_quantization_lib.cc
namespace stablehlo::quantization::pywrap { using ::mlir::quant::stablehlo::QuantizeStaticRangePtq; using ::mlir::quant::stablehlo::QuantizeWeightOnlyPtq; using ::tensorflow::SignatureDef; using ::tensorflow::quantization::PyFunctionLibrary; // Note for maintainers: the definitions should ONLY mirror existing functions // defined in different targets. Do not include any extra business logic that
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 02:51:53 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/python/pywrap_quantization_lib.h
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>& signature_def_map, const tensorflow::quantization::PyFunctionLibrary& py_function_library); // Function used by the pywrap_quantization module to mirror // `::mlir::quant::stablehlo::QuantizeWeightOnlyPtq`.
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/stablehlo/cc/static_range_ptq.h
py_function_library, absl::string_view src_saved_model_path, std::vector<std::string> signature_keys, std::unordered_set<std::string> tags, absl::flat_hash_map<std::string, tensorflow::SignatureDef> signature_def_map, absl::flat_hash_map<FunctionName, FunctionAlias> function_aliases); // Runs the static-range post-training quantization (PTQ) on `module_op`. absl::StatusOr<ModuleOp> Run(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 19 02:44:03 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/weight_only_ptq.cc
#include "tsl/platform/statusor.h" namespace mlir::quant::stablehlo { using ::stablehlo::quantization::GetReportFilePath; using ::stablehlo::quantization::QuantizationConfig; using ::tensorflow::SignatureDef; using ::tensorflow::quantization::ExportedModel; using ::tensorflow::quantization::PyFunctionLibrary; using ::tensorflow::quantization::RunPasses;
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
// weight-only quantization. // // `signature_keys` specify the signatures that correspond to functions to be // quantized. `signature_def_map` connects the signature keys to // `SignatureDef`s. // // Returns a non-OK status when the quantization is not successful. // LINT.IfChange absl::Status QuantizeWeightOnlyPtq( absl::string_view src_saved_model_path, absl::string_view dst_saved_model_path,
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/quantization/stablehlo/python/pywrap_quantization.cc
`dst_saved_model_path`. The user should pass a serialized `QuantizationConfig` for the `quantization_config_serialized` argument, and a signature key -> serialized `SignatureDef` mapping for the `signature_def_map_serialized` argument. Raises `StatusNotOk` exception if when the run was unsuccessful. )pbdoc",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 13:51:40 UTC 2024 - 5.1K bytes - Viewed (0)