Searched refs:dynamic_batch (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | api.cc | 339 options.dynamic_batch = dynamic_batch_; in Serialize()
353 dynamic_batch_ = options.dynamic_batch; in OnOptions()
359 void set_dynamic_batch(bool dynamic_batch) { dynamic_batch_ = dynamic_batch; } in set_dynamic_batch() argument
393 compiled_model_impl->set_dynamic_batch(options.dynamic_batch); in Compile()
|
| D | compiler_options.h | 57 bool dynamic_batch = false; member
|
| D | serialization.h | 39 bool dynamic_batch = false; member
|
| D | serialization_test.cc | 190 options.dynamic_batch = true; in TEST()
204 EXPECT_TRUE(handler.options.dynamic_batch); in TEST()
|
| D | serialization.cc | 315 params_builder.add_dynamic_batch(options.dynamic_batch); in Encode()
571 options.dynamic_batch = fb_parameters.dynamic_batch(); in ParseParameters()
|
| D | compiled_model.fbs | 166 dynamic_batch:bool;
|
| D | compiler.cc | 115 if (options_.dynamic_batch) { in Compile()
|
| /external/tensorflow/tensorflow/python/keras/saving/saved_model/ |
| D | model_serialization.py | 39 …metadata['save_spec'] = self.obj._get_save_spec(dynamic_batch=False) # pylint: disable=protected-…
|
| D | revive_test.py | 390 model._get_save_spec(dynamic_batch=False),
391 revived._get_save_spec(dynamic_batch=False))
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | tflite_keras_util.py | 79 …input_specs = model._get_save_spec(dynamic_batch=not keep_original_batch_size) # pylint: disable=…
|
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | tf_utils.py | 461 def get_tensor_spec(t, dynamic_batch=False, name=None): argument
477 if not dynamic_batch:
|
| /external/tensorflow/tensorflow/python/keras/saving/ |
| D | saving_utils.py | 79 …input_specs = model._get_save_spec(dynamic_batch=not keep_original_batch_size) # pylint: disable=…
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | functional.py | 879 def _get_save_spec(self, dynamic_batch=True): argument
883 dynamic_batch = False
884 return super(Functional, self)._get_save_spec(dynamic_batch)
|
| D | base_layer.py | 3082 def _get_save_spec(self, dynamic_batch=True): argument
3087 lambda t: tf_utils.get_tensor_spec(t, dynamic_batch=dynamic_batch),
|
| D | training.py | 1545 t, tf_utils.get_tensor_spec(t, dynamic_batch=True).shape)
2496 tf_utils.get_tensor_spec(tensor, dynamic_batch=False, name=name))
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/ |
| D | gl_delegate.cc | 257 compile_options.dynamic_batch = in Prepare()
|