- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 35 for py_function_library (0.3 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/python/pywrap_quantize_model.cc
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; tags.insert(quantization_options.tags().begin(),
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
using ::tensorflow::SignatureDef; using ::tensorflow::quantization::ExportedModel; using ::tensorflow::quantization::PyFunctionLibrary; StaticRangePtqComponent::StaticRangePtqComponent( absl::Nonnull<MLIRContext*> ctx, absl::Nonnull<const PyFunctionLibrary*> py_function_library, const absl::string_view src_saved_model_path, std::vector<std::string> signature_keys, std::unordered_set<std::string> 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.pyi
py_function_library: py_function_lib.PyFunctionLibrary, ) -> Any: ... # Status # LINT.ThenChange() # LINT.IfChange(quantize_ptq_static_range) def quantize_ptq_static_range( 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) -
tensorflow/compiler/mlir/quantization/tensorflow/python/pywrap_quantize_model_test.py
dst_saved_model_path, quant_opts_serialized, signature_keys=signature_def_keys, signature_def_map_serialized={}, function_aliases={}, py_function_library=py_function_lib.PyFunctionLibrary(), representative_dataset_file_map_serialized=None, ) def test_quantize_model_fails_when_invalid_quant_options_type(self): src_saved_model_path = self.create_tempdir().full_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/pywrap_quantization.pyi
src_saved_model_path: str, dst_saved_model_path: str, quantization_config_serialized: bytes, *, signature_keys: list[str], signature_def_map_serialized: dict[str, bytes], py_function_library: py_function_lib.PyFunctionLibrary, ) -> Any: ... # Status # LINT.ThenChange() # LINT.IfChange(weight_only_ptq) def weight_only_ptq( src_saved_model_path: str, dst_saved_model_path: str,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 13:51:40 UTC 2024 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/python/pywrap_quantization_lib.cc
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( absl::string_view src_saved_model_path,
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/cc/calibration/statistics.cc
const CalibrationOptions& calibration_options, const PyFunctionLibrary& py_function_library) { TF_ASSIGN_OR_RETURN(const CalibrationStatisticsFlatMap statistics_map, ReadStatistics(calibration_data_dir)); absl::Status status = absl::OkStatus(); module_op.walk([&py_function_library, &calibration_options, &status,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/python/quantization.py
quantization_config_serialized=config.SerializeToString(), signature_keys=list(signature_def_map.keys()), signature_def_map_serialized=signature_def_map_serialized, py_function_library=py_function_lib.PyFunctionLibrary(), ) elif _has_quantization_method(config.specs, 'weight_only_ptq'): pywrap_quantization.weight_only_ptq( src_saved_model_path, dst_saved_model_path,
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/quantization/stablehlo/python/pywrap_quantization_lib.h
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`. absl::Status PywrapQuantizeWeightOnlyPtq( absl::string_view src_saved_model_path,
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/weight_only_ptq.cc
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(), quantization_config.tf_saved_model().tags().end());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 02:59:01 UTC 2024 - 5.1K bytes - Viewed (0)