Home
last modified time | relevance | path

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

/cmsis-nn-latest/Tests/UnitTest/RefactoredTestGen/Lib/
Dop_conv.py137 x_output = output_details[0]['shape'][2]
140 def calculate_padding(x_output, y_output, params): argument
149 pad_along_width = max((x_output - 1) * params["stride_x"] + filter_x - x_input, 0)
169 … pad_y_with_offset, pad_x_with_offset, pad_y, pad_x = calculate_padding(x_output, y_output, params)
206 generated_params["output_w"] = x_output
207 generated_params["dst_size"] = x_output * y_output * params["out_ch"] * params["batch_size"]
/cmsis-nn-latest/Tests/UnitTest/
Dmodel_extractor.py100 … self.x_output * self.y_output * self.output_ch * self.batches))
121 f.write("#define {}_OUTPUT_W {}\n".format(prefix, self.x_output))
155 [self.y_output, self.x_output] = output_shape
164 [self.y_output, self.x_output] = output_shape
166 [d, self.y_output, self.x_output, d1] = output_shape
168 self.calculate_padding(self.x_output, self.y_output, self.x_input, self.y_input)
Dpooling_settings.py110 self.x_output = output_details[0]['shape'][2]
113 self.calculate_padding(self.x_output, self.y_output, self.x_input, self.y_input)
Dsoftmax_settings.py60 self.x_input = self.x_output = x_in
104 f.write("#define {}_DST_SIZE {}\n".format(prefix, self.x_output * self.y_output))
Dtest_settings.py330 f.write("#define {}_OUTPUT_W {}\n".format(prefix, self.x_output))
354 prefix, self.x_output * self.y_output * self.output_ch * self.batches))
401 def calculate_padding(self, x_output, y_output, x_input, y_input): argument
407 pad_along_width = max((x_output - 1) * self.stride_x + filter_x - x_input, 0)
Dconv_settings.py413 self.x_output = output_details[0]['shape'][2]
417 self.calculate_padding(self.x_input, self.y_input, self.x_output, self.y_output)
419 self.calculate_padding(self.x_output, self.y_output, self.x_input, self.y_input)
Dadd_mul_settings.py68 self.x_input = self.x_output = x_in
Dfully_connected_settings.py272 self.x_output = 1