- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for _create_while_model (0.37 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test_base.py
signature_key, tags, inputs={input_key: in_placeholder}, outputs={output_key: output_tensor}, ) return in_placeholder def _create_while_model(self, input_shape: Sequence[int] = (1, 32, 32, 512)): class WhileModel(module.Module): """A model with a while op.""" def __init__(self):
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/integration_test/quantize_model_test.py
self.assertAllClose(original_outputs, quantized_output, atol=442.7) @test_util.run_v2_only def test_while_op_model( self, ): input_shape = (1, 5, 5, 32) model = self._create_while_model(input_shape) saved_model_save.save(model, self._input_saved_model_path) tags = {tag_constants.SERVING} quantization_options = quant_opts_pb2.QuantizationOptions(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 235.6K bytes - Viewed (0)