- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 27 for signature_def_map (0.21 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/python/quantization.py
_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: Signature def map where the values (`SignatureDef`) are serialized. """
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/tensorflow/python/pywrap_quantize_model.cc
const QuantizationOptions& quantization_options, const std::vector<std::string>& signature_keys, const absl::flat_hash_map<std::string, SignatureDef>& signature_def_map, const PyFunctionLibrary& py_function_library) -> absl::Status { // LINT.ThenChange(pywrap_quantize_model.pyi:quantize_qat_model) std::unordered_set<std::string> tags;
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
const 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, SignatureDef> signature_def_map, absl::flat_hash_map<FunctionName, FunctionAlias> function_aliases) : ctx_(ctx) { // Initialize the three sub-components. sub_components_[0] = std::make_unique<PreCalibrationComponent>(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/lite/quantization/stablehlo/quantization.cc
saved_model_bundle.GetSignatures(); absl::flat_hash_map<std::string, SignatureDef> signature_def_map( signatures.begin(), signatures.end()); // Init op is only used during initialization and it's not a target for // quantization. signature_def_map.erase(kSavedModelInitOpSignatureKey); return signature_def_map; } // Retrieves the function name -> function alias mapping from the // `SavedModelBundle`.
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/tensorflow/python/quantize_model.py
return True return False 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: Signature def map where the values (`SignatureDef`) are serialized. """
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 34.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/common_v1.py
anything else. 2. Converts signature_def_map to SavedModel V1 3. Converts SavedModel V1 to MLIR 4. Prints the textual MLIR to stdout (it is expected that the caller will have FileCheck checks in its file to check this output). This is only for use by the MLIR SavedModel importer tests. Args: create_signature: A functor that return signature_def_map, init_op and
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 20 13:19:26 UTC 2023 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/python/pywrap_quantization_lib.cc
const std::vector<std::string>& signature_keys, const absl::flat_hash_map<std::string, SignatureDef>& signature_def_map, const PyFunctionLibrary& py_function_library) { return QuantizeStaticRangePtq(src_saved_model_path, dst_saved_model_path, config, signature_keys, signature_def_map, py_function_library); } absl::Status PywrapQuantizeWeightOnlyPtq(
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/tensorflow/python/save_model.py
graph. Args: signature_def_map: the signatures to validate. exported_graph: The PTQ-exported GraphDef. Returns: The signatures with tensor names prefixed with signature name if necessary. Raises: ValueError: Iff the signatures are not consistent with the graph. """ for signature_key, signature_def in signature_def_map.items():
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 01:09:50 UTC 2024 - 12.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/weight_only_ptq.cc
const absl::string_view dst_saved_model_path, QuantizationConfig quantization_config, const std::vector<std::string>& signature_keys, const absl::flat_hash_map<std::string, SignatureDef>& signature_def_map, const PyFunctionLibrary& py_function_library) { std::unordered_set<std::string> tags; tags.insert(quantization_config.tf_saved_model().tags().begin(),
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
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)