Searched refs:dataset_shape (Results 1 – 2 of 2) sorted by relevance
/cmsis-nn-latest/Tests/UnitTest/ |
D | test_settings.py | 448 def convert_and_interpret(self, model, inttype, input_data=None, dataset_shape=None): argument 452 self.convert_model(model, inttype, dataset_shape) 455 def convert_model(self, model, inttype, dataset_shape=None, int16x8_int32bias=False): argument 461 if dataset_shape: 462 representative_dataset_shape = dataset_shape
|
D | lstm_settings.py | 174 interpreter = self.convert_and_interpret(model, tf.int8, input_data, dataset_shape=shape)
|