| /cmsis-nn-latest/Tests/UnitTest/RefactoredTestGen/Lib/ |
| D | op_conv.py | 28 def get_shapes(params): argument
32 params["stride_x"] = 1 if "stride_x" not in params else params["stride_x"]
33 params["stride_y"] = 1 if "stride_y" not in params else params["stride_y"]
34 params["dilation_x"] = 1 if "dilation_x" not in params else params["dilation_x"]
35 params["dilation_y"] = 1 if "dilation_y" not in params else params["dilation_y"]
36 params["batch_size"] = 1 if "batch_size" not in params else params["batch_size"]
37 params["generate_bias"] = True if "generate_bias" not in params else params["generate_bias"]
38 if "out_activation_min" not in params:
39 params["out_activation_min"] = Lib.op_utils.get_dtype_min(params["input_data_type"])
40 if "out_activation_max" not in params:
[all …]
|
| D | op_fully_connected.py | 25 def get_shapes(params): argument
29 params["batch_size"] = 1 if "batch_size" not in params else params["batch_size"]
30 params["generate_bias"] = True if "generate_bias" not in params else params["generate_bias"]
31 if "out_activation_min" not in params:
32 params["out_activation_min"] = Lib.op_utils.get_dtype_min(params["input_data_type"])
33 if "out_activation_max" not in params:
34 params["out_activation_max"] = Lib.op_utils.get_dtype_max(params["input_data_type"])
35 if "bias_min" not in params:
36 params["bias_min"] = Lib.op_utils.get_dtype_min("int32_t")
37 if "bias_max" not in params:
[all …]
|
| D | op_lstm.py | 28 def get_shapes(params): argument
30 if params["time_major"] and params["tflite_generator"] == "json":
31 shapes["input"] = (params["time_steps"], params["batch_size"], params["input_size"])
33 shapes["input"] = (params["batch_size"], params["time_steps"], params["input_size"])
35 shapes["input_weights"] = (params["input_size"], params["hidden_size"])
36 shapes["all_input_weights"] = (params["input_size"], params["hidden_size"] * 4)
38 shapes["hidden_weights"] = (params["hidden_size"], params["hidden_size"])
39 shapes["all_hidden_weights"] = (params["hidden_size"], params["hidden_size"] * 4)
41 shapes["bias"] = (1, params["hidden_size"])
42 shapes["all_bias"] = (params["hidden_size"] * 4, )
[all …]
|
| D | op_utils.py | 24 def __init__(self, params, tensors, scales, effective_scales, aliases={}): argument
25 … self.params = params # All other params which are generated rather than given in the test-plan
41 def generate_keras_model(output_path, shapes, params): argument
46 def generate_data_tflite(tflite_path, params) -> Generated_data: argument
54 def generate_data_json(shapes, params) -> Generated_data: argument
70 def get_dtype(name, params): argument
72 return params["bias_data_type"]
74 return params["weights_data_type"]
76 return params["shift_and_mult_data_type"]
78 return params["input_data_type"]
|
| D | test.py | 49 def generate(params, args, fpaths): argument
53 if (params["interpreter"] == "tflite_runtime") and (not tflite_runtime_imported):
56 if (params["interpreter"] == "tflite_micro") and (not tflite_micro_imported):
60 op_type = get_op_type(params["op_type"])
61 shapes = op_type.get_shapes(params)
64 fpaths["data_folder"] = pathlib.Path("TestCases") / "TestData" / params["name"]
70 if params["tflite_generator"] == "keras":
71 keras_model = op_type.generate_keras_model(shapes, params)
75 dtype=params["input_data_type"],
76 bias_dtype=params["bias_data_type"],
[all …]
|
| D | test_suite.py | 21 def generate(params, args): argument
28 common_test_params = {key: val for key, val in params.items() if is_common(key)}
37 for test_params in params["tests"]:
|
| /cmsis-nn-latest/Source/NNSupportFunctions/ |
| D | arm_nn_lstm_step_s16.c | 49 const cmsis_nn_lstm_params *params, in arm_nn_lstm_step_s16() argument
61 …arm_nn_lstm_calculate_gate_s16(data_in, hidden_in, ¶ms->forget_gate, params, forget_gate, batc… in arm_nn_lstm_step_s16()
70 params->forget_to_cell_multiplier, in arm_nn_lstm_step_s16()
71 params->forget_to_cell_shift, in arm_nn_lstm_step_s16()
74 params->hidden_size * params->batch_size); in arm_nn_lstm_step_s16()
76 …arm_nn_lstm_calculate_gate_s16(data_in, hidden_in, ¶ms->input_gate, params, input_gate, batch_… in arm_nn_lstm_step_s16()
78 …arm_nn_lstm_calculate_gate_s16(data_in, hidden_in, ¶ms->cell_gate, params, cell_gate, batch_of… in arm_nn_lstm_step_s16()
87 params->input_to_cell_multiplier, in arm_nn_lstm_step_s16()
88 params->input_to_cell_shift, in arm_nn_lstm_step_s16()
89 -params->cell_clip, in arm_nn_lstm_step_s16()
[all …]
|
| D | arm_nn_lstm_step_s8.c | 48 const cmsis_nn_lstm_params *params, in arm_nn_lstm_step_s8() argument
60 …arm_nn_lstm_calculate_gate_s8_s16(data_in, hidden_in, ¶ms->forget_gate, params, forget_gate, b… in arm_nn_lstm_step_s8()
69 params->forget_to_cell_multiplier, in arm_nn_lstm_step_s8()
70 params->forget_to_cell_shift, in arm_nn_lstm_step_s8()
73 params->hidden_size * params->batch_size); in arm_nn_lstm_step_s8()
75 …arm_nn_lstm_calculate_gate_s8_s16(data_in, hidden_in, ¶ms->input_gate, params, input_gate, bat… in arm_nn_lstm_step_s8()
76 …arm_nn_lstm_calculate_gate_s8_s16(data_in, hidden_in, ¶ms->cell_gate, params, cell_gate, batch… in arm_nn_lstm_step_s8()
85 params->input_to_cell_multiplier, in arm_nn_lstm_step_s8()
86 params->input_to_cell_shift, in arm_nn_lstm_step_s8()
87 -params->cell_clip, in arm_nn_lstm_step_s8()
[all …]
|
| D | arm_nn_lstm_calculate_gate_s16.c | 50 const cmsis_nn_lstm_params *params, in arm_nn_lstm_calculate_gate_s16() argument
55 memset(output, 0, params->hidden_size * params->batch_size * sizeof(int16_t)); in arm_nn_lstm_calculate_gate_s16()
63 params->input_size, in arm_nn_lstm_calculate_gate_s16()
64 params->hidden_size, in arm_nn_lstm_calculate_gate_s16()
65 params->batch_size, in arm_nn_lstm_calculate_gate_s16()
77 params->hidden_size, in arm_nn_lstm_calculate_gate_s16()
78 params->hidden_size, in arm_nn_lstm_calculate_gate_s16()
79 params->batch_size, in arm_nn_lstm_calculate_gate_s16()
83 …arm_nn_activation_s16(output, output, params->hidden_size * params->batch_size, 0, gate->activatio… in arm_nn_lstm_calculate_gate_s16()
|
| D | arm_nn_lstm_calculate_gate_s8_s16.c | 57 const cmsis_nn_lstm_params *params, in arm_nn_lstm_calculate_gate_s8_s16() argument
62 memset(output, 0, params->hidden_size * params->batch_size * sizeof(int16_t)); in arm_nn_lstm_calculate_gate_s8_s16()
70 params->input_size, in arm_nn_lstm_calculate_gate_s8_s16()
71 params->hidden_size, in arm_nn_lstm_calculate_gate_s8_s16()
72 params->batch_size, in arm_nn_lstm_calculate_gate_s8_s16()
83 params->hidden_size, in arm_nn_lstm_calculate_gate_s8_s16()
84 params->hidden_size, in arm_nn_lstm_calculate_gate_s8_s16()
85 params->batch_size, in arm_nn_lstm_calculate_gate_s8_s16()
89 …arm_nn_activation_s16(output, output, params->hidden_size * params->batch_size, 0, gate->activatio… in arm_nn_lstm_calculate_gate_s8_s16()
|
| /cmsis-nn-latest/Source/LSTMFunctions/ |
| D | arm_lstm_unidirectional_s16.c | 51 const cmsis_nn_lstm_params *params, in arm_lstm_unidirectional_s16() argument
56 memset(buffers->cell_state, 0, params->batch_size * params->hidden_size * sizeof(int16_t)); in arm_lstm_unidirectional_s16()
57 if (params->time_major) in arm_lstm_unidirectional_s16()
60 for (int t = 0; t < params->time_steps; t++) in arm_lstm_unidirectional_s16()
62 const int16_t *data_in = input + (t * params->batch_size * params->input_size); in arm_lstm_unidirectional_s16()
63 int16_t *hidden_out = output + (t * params->batch_size * params->hidden_size); in arm_lstm_unidirectional_s16()
64 …_cmsis_nn_status status = arm_nn_lstm_step_s16(data_in, hidden_in, hidden_out, params, buffers, 1); in arm_lstm_unidirectional_s16()
76 for (int t = 0; t < params->time_steps; t++) in arm_lstm_unidirectional_s16()
78 const int16_t *data_in = input + (t * params->input_size); in arm_lstm_unidirectional_s16()
79 int16_t *hidden_out = output + (t * params->hidden_size); in arm_lstm_unidirectional_s16()
[all …]
|
| D | arm_lstm_unidirectional_s8.c | 51 const cmsis_nn_lstm_params *params, in arm_lstm_unidirectional_s8() argument
56 memset(buffers->cell_state, 0, params->batch_size * params->hidden_size * sizeof(int16_t)); in arm_lstm_unidirectional_s8()
57 if (params->time_major) in arm_lstm_unidirectional_s8()
60 for (int t = 0; t < params->time_steps; t++) in arm_lstm_unidirectional_s8()
62 const int8_t *data_in = input + (t * params->batch_size * params->input_size); in arm_lstm_unidirectional_s8()
63 int8_t *hidden_out = output + (t * params->batch_size * params->hidden_size); in arm_lstm_unidirectional_s8()
64 …arm_cmsis_nn_status status = arm_nn_lstm_step_s8(data_in, hidden_in, hidden_out, params, buffers, … in arm_lstm_unidirectional_s8()
76 for (int t = 0; t < params->time_steps; t++) in arm_lstm_unidirectional_s8()
78 const int8_t *data_in = input + (t * params->input_size); in arm_lstm_unidirectional_s8()
79 int8_t *hidden_out = output + (t * params->hidden_size); in arm_lstm_unidirectional_s8()
[all …]
|
| /cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_lstm_unidirectional_s8/ |
| D | test_arm_lstm_unidirectional_s8.c | 161 const cmsis_nn_lstm_params params = {LSTM_1_TIME_MAJOR, in lstm_1_arm_lstm_unidirectional_s8() local
187 … arm_cmsis_nn_status result = arm_lstm_unidirectional_s8(lstm_1_input, output, ¶ms, &buffers); in lstm_1_arm_lstm_unidirectional_s8()
312 const cmsis_nn_lstm_params params = {LSTM_2_TIME_MAJOR, in lstm_2_arm_lstm_unidirectional_s8() local
339 … arm_cmsis_nn_status result = arm_lstm_unidirectional_s8(lstm_2_input, output, ¶ms, &buffers); in lstm_2_arm_lstm_unidirectional_s8()
465 const cmsis_nn_lstm_params params = {LSTM_ONE_TIME_STEP_TIME_MAJOR, in lstm_one_time_step_arm_lstm_unidirectional_s8() local
491 …nn_status result = arm_lstm_unidirectional_s8(lstm_one_time_step_input, output, ¶ms, &buffers); in lstm_one_time_step_arm_lstm_unidirectional_s8()
|
| /cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_lstm_unidirectional_s16/ |
| D | test_arm_lstm_unidirectional_s16.c | 152 const cmsis_nn_lstm_params params = {lstm_1_s16_time_major, in lstm_1_s16() local
177 …arm_cmsis_nn_status result = arm_lstm_unidirectional_s16(lstm_1_s16_input, output, ¶ms, &buffe… in lstm_1_s16()
298 const cmsis_nn_lstm_params params = {lstm_2_s16_time_major, in lstm_2_s16() local
323 …arm_cmsis_nn_status result = arm_lstm_unidirectional_s16(lstm_2_s16_input, output, ¶ms, &buffe… in lstm_2_s16()
446 const cmsis_nn_lstm_params params = {lstm_one_time_step_s16_time_major, in lstm_one_time_step_s16() local
471 …atus result = arm_lstm_unidirectional_s16(lstm_one_time_step_s16_input, output, ¶ms, &buffers); in lstm_one_time_step_s16()
|
| /cmsis-nn-latest/Include/ |
| D | arm_nnsupportfunctions.h | 1747 const cmsis_nn_lstm_params *params,
1771 const cmsis_nn_lstm_params *params,
1792 const cmsis_nn_lstm_params *params,
1813 const cmsis_nn_lstm_params *params,
|
| D | arm_nnfunctions.h | 2541 const cmsis_nn_lstm_params *params,
2562 const cmsis_nn_lstm_params *params,
|
| /cmsis-nn-latest/Documentation/Doxygen/style_template/ |
| D | extra_stylesheet.css | 1117 .params, .retval, .exception, .tparams {
1122 .params .paramname, .retval .paramname {
1127 .params .paramtype {
1132 .params .paramdir {
|
