Searched refs:keep_original_batch_size (Results 1 – 3 of 3) sorted by relevance
58 def model_input_signature(model, keep_original_batch_size=False): argument79 …input_specs = model._get_save_spec(dynamic_batch=not keep_original_batch_size) # pylint: disable=…
893 self._keras_model, keep_original_batch_size=True)