Home
last modified time | relevance | path

Searched refs:tensor_details (Results 1 – 6 of 6) sorted by relevance

/cmsis-nn-latest/Tests/UnitTest/RefactoredTestGen/Lib/
Dop_maximum_minimum.py69 tensor_details = interpreter.get_tensor_details()
71 input_1 = tensor_details[0]
72 input_2 = tensor_details[1]
Dop_batch_matmul.py63 tensor_details = interpreter.get_tensor_details()
65 lhs = tensor_details[0]
66 rhs = tensor_details[1]
Dop_lstm.py86 tensor_details = interpreter.get_tensor_details()
93 input_state = tensor_details[0]
95 cell_state = tensor_details[14 + time_major_offset * 2]
97 output_state = tensor_details[13 + time_major_offset * 2]
109 detail = tensor_details[tensor_index + time_major_offset]
130 generated_params["output_zero_point"] = tensor_details[20 + time_major_offset *
Dop_fully_connected.py156 tensor_details = interpreter.get_tensor_details()
164 filter_layer = tensor_details[filter_index]
168 bias_layer = tensor_details[bias_index]
Dop_conv.py57 tensor_details = interpreter.get_tensor_details()
66 filter_layer = tensor_details[filter_index]
70 bias_layer = tensor_details[bias_index]
/cmsis-nn-latest/Tests/UnitTest/
Dmodel_extractor.py170 def extract_from_model(self, json_file, tensor_details): argument
209 input_tensor = tensor_details[input_index]
210 output_tensor = tensor_details[output_index]
224 weight_tensor = tensor_details[weights_index]
280 tensor_details = interpreter.get_tensor_details()
295 self.extract_from_model(json_file, tensor_details)