Searched refs:y_output (Results 1 – 8 of 8) sorted by relevance
| /cmsis-nn-latest/Tests/UnitTest/RefactoredTestGen/Lib/ |
| D | op_conv.py | 45 y_output = interpreter.get_output_details(0)["shape"][1]
81 y_output = output_details[0]['shape'][1]
89 def calculate_padding(x_output, y_output, params): argument
99 pad_along_height = max((y_output - 1) * params["stride_y"] + filter_y - y_input, 0)
119 … pad_y_with_offset, pad_x_with_offset, pad_y, pad_x = calculate_padding(x_output, y_output, params)
124 generated_params["output_h"] = y_output
126 generated_params["dst_size"] = x_output * y_output * params["out_ch"] * params["batch_size"]
|
| /cmsis-nn-latest/Tests/UnitTest/ |
| D | model_extractor.py | 100 … self.x_output * self.y_output * self.output_ch * self.batches))
122 f.write("#define {}_OUTPUT_H {}\n".format(prefix, self.y_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)
|
| D | pooling_settings.py | 111 self.y_output = output_details[0]['shape'][1]
113 self.calculate_padding(self.x_output, self.y_output, self.x_input, self.y_input)
|
| D | softmax_settings.py | 61 self.y_input = self.y_output = y_in
104 f.write("#define {}_DST_SIZE {}\n".format(prefix, self.x_output * self.y_output))
|
| D | test_settings.py | 331 f.write("#define {}_OUTPUT_H {}\n".format(prefix, self.y_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
408 pad_along_height = max((y_output - 1) * self.stride_y + filter_y - y_input, 0)
|
| D | conv_settings.py | 414 self.y_output = output_details[0]['shape'][1]
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)
|
| D | add_mul_settings.py | 69 self.y_input = self.y_output = y_in
|
| D | fully_connected_settings.py | 273 self.y_output = 1
|
