- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for py_function_library (0.48 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/tensorflow/python/quantize_model.cc
TF_ASSIGN_OR_RETURN(std::string calibration_saved_model_dir, CreateTmpDir()); py_function_library.SaveExportedModel( calibration_saved_model_dir, calibration_exported_model, saved_model_path, tags, signature_def_map); py_function_library.RunCalibration( calibration_saved_model_dir, signature_keys, tags, quantization_options.force_graph_mode_calibration(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/quantize_model.py
signature_keys=list(quant_opts.signature_keys), signature_def_map_serialized=_serialize_signature_def_map( signature_def_map ), py_function_library=py_function_lib.PyFunctionLibrary(), ) def _run_static_range_ptq( src_saved_model_path: str, dst_saved_model_path: str, quant_opts: _QuantizationOptions,
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/quantization/tensorflow/python/py_function_lib.py
# comments in `NotNoneT`. return True class PyFunctionLibrary(pywrap_function_lib.PyFunctionLibrary): """Wrapper class for overridden python method definitions. This class contains python methods that overrides C++ virtual functions declared in `pywrap_function_lib.PyFunctionLibrary`. """ # LINT.IfChange(save_exported_model) def save_exported_model( self,
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/lite/python/saved_model_to_tfl_flatbuffer.cc
#include "tensorflow/lite/toco/toco_flags.pb.h" #include "tensorflow/lite/toco/types.pb.h" #include "tsl/platform/errors.h" #include "tsl/platform/statusor.h" namespace tensorflow { using tensorflow::quantization::PyFunctionLibrary; Status HandleInputOutputArraysWithModule( const toco::ModelFlags& model_flags, mlir::OwningOpRef<mlir::ModuleOp>* module) { mlir::func::FuncOp entry_function = nullptr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:39:37 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc
namespace tensorflow { namespace { using mlir::MLIRContext; using mlir::ModuleOp; using mlir::Operation; using mlir::OwningOpRef; using ::stablehlo::quantization::QuantizationConfig; using ::tensorflow::quantization::PyFunctionLibrary; bool IsControlFlowV1Op(Operation* op) { return mlir::isa<mlir::tf_executor::SwitchOp, mlir::tf_executor::MergeOp, mlir::tf_executor::EnterOp, mlir::tf_executor::ExitOp,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/python/tf_tfl_flatbuffer_helpers.cc
#include "tsl/platform/errors.h" #include "tsl/platform/protobuf.h" // IWYU pragma: keep #include "tsl/platform/statusor.h" namespace tensorflow { namespace internal { namespace { using ::tensorflow::quantization::PyFunctionLibrary; using ::tflite::optimize::ReducedPrecisionSupport; // Op def string for TFLite_Detection_PostProcess Op. constexpr mlir::StringRef kDetectionPostProcessOp =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:39:37 UTC 2024 - 17.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/python/converter_python_api.cc
PyObject* input_contents_txt_raw, bool extended_return, PyObject* debug_info_txt_raw, bool enable_mlir_converter, const tensorflow::quantization::PyFunctionLibrary* quantization_py_function_library) { // Use Python C API to validate and convert arguments. In py3 (bytes), // in py2 (str). auto ConvertArg = [&](PyObject* obj, bool* error) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 19.2K bytes - Viewed (0)