Home
last modified time | relevance | path

Searched refs:input_shape (Results 1 – 8 of 8) sorted by relevance

/cmsis-nn-latest/Tests/UnitTest/
Dpooling_settings.py89 input_shape = (self.batches, self.y_input, self.x_input, self.input_ch)
90 model.add(keras.layers.InputLayer(input_shape=input_shape[1:], batch_size=self.batches))
96 input_shape=input_shape[1:]))
102 input_shape=input_shape[1:]))
Dsoftmax_settings.py106 def get_softmax_randomized_input_data(self, input_data, input_shape): argument
109 input_data = tf.reshape(input_data, input_shape)
111 input_data = self.get_randomized_data(input_shape,
151 input_shape = (self.y_input, self.x_input)
152 model.add(keras.layers.Softmax(input_shape=input_shape))
Dmodel_extractor.py144 def shape_to_config(self, input_shape, filter_shape, output_shape, layer_name): argument
146 [_, self.filter_y, self.filter_x, _] = input_shape
149 [self.batches, self.y_input, self.x_input, self.input_ch] = input_shape
153 [self.batches, self.input_ch] = input_shape
160 [self.y_input, self.x_input] = input_shape
162 if len(input_shape) == 4:
216 input_shape = input_tensor['shape']
249 self.shape_to_config(input_shape, filter_shape, output_shape, builtin_name)
288 input_shape = input_details[0]['shape']
289 input_data = self.get_randomized_input_data(input_data, input_shape)
Dadd_mul_settings.py75 input_shape = (1, self.y_input, self.x_input, self.input_ch)
77 input_data1 = self.get_randomized_data(list(input_shape),
81 input_data2 = self.get_randomized_data(list(input_shape),
94 input1 = keras.layers.Input(shape=input_shape[1:])
95 input2 = keras.layers.Input(shape=input_shape[1:])
Dconv_settings.py335 input_shape = (self.batches, self.y_input, self.x_input, self.input_ch)
336 model.add(keras.layers.InputLayer(input_shape=input_shape[1:], batch_size=self.batches))
342 input_shape=input_shape[1:],
356 input_shape=input_shape[1:],
369 input_shape=input_shape[1:],
Dtest_settings.py272 def get_randomized_input_data(self, input_data, input_shape=None): argument
274 if input_shape is None:
275 input_shape = [self.batches, self.y_input, self.x_input, self.input_ch]
277 input_data = tf.reshape(input_data, input_shape)
279 input_data = self.get_randomized_data(input_shape,
Dfully_connected_settings.py241 keras.layers.InputLayer(input_shape=(self.y_input * self.x_input * self.input_ch, ),
/cmsis-nn-latest/Tests/UnitTest/RefactoredTestGen/Lib/
Dop_conv.py75 input_shape = (params["batch_size"], params["input_h"], params["input_w"], params["in_ch"])
76 … model.add(keras.layers.InputLayer(input_shape=input_shape[1:], batch_size=params["batch_size"]))
82 input_shape=input_shape[1:],