- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 22 for SignatureDef (0.38 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 (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/tensorflow/python/pywrap_quantize_model.cc
#include "tensorflow/core/framework/attr_value.pb.h" #include "tensorflow/core/protobuf/meta_graph.pb.h" namespace py = pybind11; namespace { using ::tensorflow::SignatureDef; using ::tensorflow::quantization::ExportedModel; using ::tensorflow::quantization::PyFunctionLibrary; using ::tensorflow::quantization::QuantizationOptions; using ::tensorflow::quantization::QuantizeDynamicRangePtq;
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
#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/tensorflow/python/save_model.py
# Mapping of signature def key -> SignatureDef. _SignatureDefMap = Mapping[str, meta_graph_pb2.SignatureDef] def get_signatures_from_saved_model( saved_model_path: str, signature_keys: Optional[Sequence[str]] = None, tags: Optional[Collection[str]] = None, ) -> Dict[str, meta_graph_pb2.SignatureDef]: """Gets a map from signature keys to their SignatureDef. Args:
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/tensorflow/python/py_function_lib.py
sess: session.Session, signature_def: meta_graph_pb2.SignatureDef, representative_dataset: rd.RepresentativeDataset, ) -> None: """Runs the representative dataset through a function for calibration. NOTE: This is intended to be run in graph mode (TF1). The function is identified by the SignatureDef. Args: sess: The Session object to run the function in.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 05:32:11 UTC 2024 - 27.4K 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/tensorflow/python/py_function_lib.h
const ExportedModel& exported_model, absl::string_view src_saved_model_path, const std::unordered_set<std::string>& tags, const absl::flat_hash_map<std::string, tensorflow::SignatureDef>& signature_def_map) const = 0; // LINT.ThenChange( // pywrap_function_lib.pyi:save_exported_model, // py_function_lib.py:save_exported_model, // )
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 09 06:33:29 UTC 2024 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/pywrap_function_lib.cc
#include "tensorflow/core/protobuf/meta_graph.pb.h" namespace py = ::pybind11; namespace { using ::stablehlo::quantization::CalibrationOptions; using ::stablehlo::quantization::MinMaxValue; using ::tensorflow::SignatureDef; using ::tensorflow::calibrator::CalibrationStatistics; using ::tensorflow::quantization::ExportedModel; using ::tensorflow::quantization::PyFunctionLibrary; using ::tensorflow::quantization::RepresentativeDatasetFile;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 09 06:33:29 UTC 2024 - 5K bytes - Viewed (0)