- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 54 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 (0) -
tensorflow/cc/tools/freeze_saved_model_test.cc
EXPECT_EQ(actual.ShortDebugString(), expected.ShortDebugString()); } // Builds a SignatureDef with the provided `inputs` and `outputs`. SignatureDef BuildSignatureDef(const std::unordered_set<string>& inputs, const std::unordered_set<string>& outputs) { SignatureDef signature_def; for (const string& input : inputs) { (*signature_def.mutable_inputs())[input].set_name(input);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 07 13:30:31 UTC 2022 - 21.7K bytes - Viewed (0) -
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/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/cc/saved_model/signature_constants.h
/// Classification API constants. /// Classification inputs. static constexpr char kClassifyInputs[] = "inputs"; /// Classification method name used in a SignatureDef. static constexpr char kClassifyMethodName[] = "tensorflow/serving/classify"; /// Classification classes output. static constexpr char kClassifyOutputClasses[] = "classes"; /// Classification scores output.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 24 22:31:28 UTC 2018 - 2.5K 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/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)