- Sort Score
- Result 10 results
- Languages All
Results 21 - 26 of 26 for save_model_path (0.18 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/python/quantize_model.py
_DYNAMIC_RANGE_DEFAULT_MIN_NUM_ELEMENTS_FOR_WEIGHTS = 1024 def _is_qat_saved_model(saved_model_path: str): """Checks if the SavedModel is QAT-enabled by looking for 'FakeQuant' ops.""" saved_model_proto = saved_model_loader.parse_saved_model(saved_model_path) for meta_graph in saved_model_proto.meta_graphs: if any( node.op.startswith('FakeQuant') for node in meta_graph.graph_def.node
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/integration_test/quantize_model_test_base.py
""" for _ in range(num_examples): yield {input_key: random_ops.random_uniform(shape, minval, maxval, dtype)} def _save_tf1_model( self, sess: session.Session, saved_model_path: str, signature_key: str, tags: Collection[str], inputs: Mapping[str, core.Tensor], outputs: Mapping[str, core.Tensor], init_op: Optional[ops.Operation] = None,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 21 08:51:46 UTC 2024 - 51.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/save_model.py
_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: saved_model_path: Path to the saved model.
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/integration_test/quantize_model_test.py
# (specified by `kDefaultConstantSizeThresholdInBytes`) for unfreezing. self._create_matmul_model( input_shape=(1, 20), weight_shape=(20, 4096), saved_model_path=self._input_saved_model_path, ) repr_ds = self._create_data_generator( input_key='input_tensor', shape=(1, 20), num_examples=2 ) tags = {tag_constants.SERVING}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 235.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/pywrap_quantize_model.cc
the function matching the signature key. Raises `StatusNotOk` exception if when the run was unsuccessful. )pbdoc", py::arg("saved_model_path"), py::arg("dst_saved_model_path"), py::arg("quantization_options_serialized"), py::kw_only(), py::arg("signature_keys"), py::arg("signature_def_map_serialized"), py::arg("py_function_library"),
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/python/integration_test/quantize_model_test.py
), 1 / 3, ) def test_when_preset_not_srq_raises_error(self): self._create_matmul_model( input_shape=(1, 1024), weight_shape=(1024, 3), saved_model_path=self._input_saved_model_path, ) config = qc.QuantizationConfig() with self.assertRaisesRegex(ValueError, 'only supports static-range PTQ'): quantization.quantize_saved_model(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 51.4K bytes - Viewed (0)