Home
last modified time | relevance | path

Searched refs:n_input (Results 1 – 25 of 58) sorted by relevance

123

/packages/modules/NeuralNetworks/common/
DQuantUtils.cpp12 int n_batch, int n_input, int16_t* output) { in ApplyLayerNorm() argument
17 for (int j = 0; j < n_input; ++j) { in ApplyLayerNorm()
18 const int32_t index = i * n_input + j; in ApplyLayerNorm()
23 int32_t mean = static_cast<int32_t>(static_cast<int64_t>(sum) * 1024 / n_input); in ApplyLayerNorm()
25 int32_t temp = kOverflowGuard / n_input; in ApplyLayerNorm()
36 for (int j = 0; j < n_input; ++j) { in ApplyLayerNorm()
37 const int32_t index = i * n_input + j; in ApplyLayerNorm()
86 void ApplySigmoid(const int16_t* input, int32_t n_batch, int32_t n_input, int16_t* output) { in ApplySigmoid() argument
88 for (int c = 0; c < n_input; c++) { in ApplySigmoid()
91 const int index = batch * n_input + c; in ApplySigmoid()
[all …]
DQuantUtils.h36 int32_t shift, int32_t n_batch, int32_t n_input, in MatrixBatchVectorMultiplyAccumulate() argument
43 for (int col = 0; col < n_input; ++col) { in MatrixBatchVectorMultiplyAccumulate()
44 int8_t input_val = input[batch * n_input + col]; in MatrixBatchVectorMultiplyAccumulate()
45 int8_t weights_val = input_to_gate_weights[row * n_input + col]; in MatrixBatchVectorMultiplyAccumulate()
142 int n_batch, int n_input, int16_t* output);
151 void ApplySigmoid(const int16_t* input, int32_t n_batch, int32_t n_input, int16_t* output);
154 void ApplyTanh(const int16_t* input, int32_t n_batch, int32_t n_input, int16_t* output) { in ApplyTanh() argument
158 for (int i = 0; i < n_input; ++i) { in ApplyTanh()
159 const int index = batch * n_input + i; in ApplyTanh()
167 inline void ApplyTanh(int32_t integer_bits, const int16_t* input, int32_t n_batch, int32_t n_input, in ApplyTanh() argument
[all …]
/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/
Dbidirectional_sequence_lstm_float16_batch_major_merge_outputs.mod.py22 n_input = 2 variable
27 input = Input("input", "TENSOR_FLOAT16", "{{{}, {}, {}}}".format(n_batch, max_time, n_input))
30 "fw_input_to_input_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
32 "fw_input_to_forget_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
34 "fw_input_to_cell_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
36 "fw_input_to_output_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
69 "bw_input_to_input_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
71 "bw_input_to_forget_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
73 "bw_input_to_cell_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
75 "bw_input_to_output_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
[all …]
Dbidirectional_sequence_lstm_merge_outputs.mod.py22 n_input = 2 variable
27 input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input))
30 "fw_input_to_input_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
32 "fw_input_to_forget_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
34 "fw_input_to_cell_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
36 "fw_input_to_output_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
69 "bw_input_to_input_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
71 "bw_input_to_forget_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
73 "bw_input_to_cell_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
75 "bw_input_to_output_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
[all …]
Dbidirectional_sequence_lstm_float16_batch_major.mod.py21 n_input = 2 variable
26 input = Input("input", "TENSOR_FLOAT16", "{{{}, {}, {}}}".format(n_batch, max_time, n_input))
29 "fw_input_to_input_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
31 "fw_input_to_forget_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
33 "fw_input_to_cell_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
35 "fw_input_to_output_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
68 "bw_input_to_input_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
70 "bw_input_to_forget_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
72 "bw_input_to_cell_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
74 "bw_input_to_output_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
[all …]
Dbidirectional_sequence_lstm.mod.py21 n_input = 2 variable
26 input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input))
29 "fw_input_to_input_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
31 "fw_input_to_forget_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
33 "fw_input_to_cell_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
35 "fw_input_to_output_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
68 "bw_input_to_input_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
70 "bw_input_to_forget_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
72 "bw_input_to_cell_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
74 "bw_input_to_output_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
[all …]
Dbidirectional_sequence_lstm_cifg_peephole.mod.py21 n_input = 2 variable
26 input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input))
29 "fw_input_to_input_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
31 "fw_input_to_forget_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
33 "fw_input_to_cell_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
35 "fw_input_to_output_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
68 "bw_input_to_input_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
70 "bw_input_to_forget_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
72 "bw_input_to_cell_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
74 "bw_input_to_output_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
[all …]
Dbidirectional_sequence_lstm_aux_input.mod.py23 n_input = 2 variable
28 input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input))
31 "fw_input_to_input_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
33 "fw_input_to_forget_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
35 "fw_input_to_cell_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
37 "fw_input_to_output_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
70 "bw_input_to_input_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
72 "bw_input_to_forget_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
74 "bw_input_to_cell_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
76 "bw_input_to_output_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
[all …]
Dbidirectional_sequence_lstm_float16_batch_major_aux_input.mod.py24 n_input = 2 variable
29 input = Input("input", "TENSOR_FLOAT16", "{{{}, {}, {}}}".format(n_batch, max_time, n_input))
32 "fw_input_to_input_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
34 "fw_input_to_forget_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
36 "fw_input_to_cell_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
38 "fw_input_to_output_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
71 "bw_input_to_input_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
73 "bw_input_to_forget_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
75 "bw_input_to_cell_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
77 "bw_input_to_output_weights", "TENSOR_FLOAT16", "{{{}, {}}}".format(n_cell, n_input))
[all …]
Dbidirectional_sequence_lstm_norm_fw_output.mod.py22 n_input = 5 variable
27 input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input))
30 "fw_input_to_input_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
32 "fw_input_to_forget_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
34 "fw_input_to_cell_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
36 "fw_input_to_output_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
69 "bw_input_to_input_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
71 "bw_input_to_forget_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
73 "bw_input_to_cell_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
75 "bw_input_to_output_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
[all …]
Dlayer_norm_lstm.mod.py23 n_input = 5 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
30 "{%d, %d}" % (n_cell, n_input))
32 "{%d, %d}" % (n_cell, n_input))
34 "{%d, %d}" % (n_cell, n_input))
36 "{%d, %d}" % (n_cell, n_input))
195 n_input = 5 variable
199 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
204 "{%d, %d}" % (n_cell, n_input))
206 "{%d, %d}" % (n_cell, n_input))
[all …]
Dlstm2_state_float16.mod.py22 n_input = 2 variable
27 input = Input("input", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_input))
29 …o_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_input))
30 …forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_input))
31 …_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_input))
32 …output_weights = Input("input_to_output_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_input))
Dunidirectional_sequence_lstm_batch_major_peephole_projection_bias.mod.py26 n_input = 5 variable
30 input = Input("input", "TENSOR_FLOAT32", "{%d, %d, %d}" % (n_batch, max_time, n_input))
33 "{%d, %d}" % (n_cell, n_input))
35 "{%d, %d}" % (n_cell, n_input))
37 "{%d, %d}" % (n_cell, n_input))
39 "{%d, %d}" % (n_cell, n_input))
Dlstm3_state2_float16.mod.py22 n_input = 5 variable
27 input = Input("input", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_input))
29 …o_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_input))
30 …forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_input))
31 …_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_input))
32 …output_weights = Input("input_to_output_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_input))
Dlstm3_state_float16.mod.py22 n_input = 5 variable
27 input = Input("input", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_input))
29 …o_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_input))
30 …forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_input))
31 …_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_input))
32 …output_weights = Input("input_to_output_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_input))
Dlstm2_state2_float16.mod.py22 n_input = 2 variable
27 input = Input("input", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_input))
29 …o_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_input))
30 …forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_input))
31 …_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_input))
32 …output_weights = Input("input_to_output_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_input))
/packages/modules/NeuralNetworks/common/cpu_operations/
DLSTMTest.cpp78 LSTMOpModel(uint32_t n_batch, uint32_t n_input, uint32_t n_cell, uint32_t n_output, in LSTMOpModel() argument
82 : n_input_(n_input), in LSTMOpModel()
137 Input_.insert(Input_.end(), n_batch * n_input, 0.f); in LSTMOpModel()
274 const int n_input = 2; in TEST() local
279 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST()
285 {n_batch, n_input}, // input tensor in TEST()
287 {n_cell, n_input}, // input_to_input_weight tensor in TEST()
288 {n_cell, n_input}, // input_to_forget_weight tensor in TEST()
289 {n_cell, n_input}, // input_to_cell_weight tensor in TEST()
290 {n_cell, n_input}, // input_to_output_weight tensor in TEST()
[all …]
DLayerNormLSTMTest.cpp87 LayerNormLSTMOpModel(uint32_t n_batch, uint32_t n_input, uint32_t n_cell, uint32_t n_output, in LayerNormLSTMOpModel() argument
91 : n_input_(n_input), in LayerNormLSTMOpModel()
148 Input_.insert(Input_.end(), n_batch * n_input, 0.f); in LayerNormLSTMOpModel()
288 const int n_input = 5; in TEST() local
293 LayerNormLSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST()
299 {n_batch, n_input}, // input tensor in TEST()
301 {n_cell, n_input}, // input_to_input_weight tensor in TEST()
302 {n_cell, n_input}, // input_to_forget_weight tensor in TEST()
303 {n_cell, n_input}, // input_to_cell_weight tensor in TEST()
304 {n_cell, n_input}, // input_to_output_weight tensor in TEST()
[all …]
/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/
Dbidirectional_sequence_lstm_state_output.mod.py21 n_input = 2 variable
26 input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input))
29 "fw_input_to_input_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
31 "fw_input_to_forget_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
33 "fw_input_to_cell_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
35 "fw_input_to_output_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
68 "bw_input_to_input_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
70 "bw_input_to_forget_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
72 "bw_input_to_cell_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
74 "bw_input_to_output_weights", "TENSOR_FLOAT32", "{{{}, {}}}".format(n_cell, n_input))
[all …]
/packages/modules/NeuralNetworks/runtime/test/specs/V1_0/
Dlstm2_state.mod.py22 n_input = 2 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
29 …o_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
30 …forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
31 …_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
32 …output_weights = Input("input_to_output_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
Dlstm2_state2.mod.py22 n_input = 2 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
29 …o_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
30 …forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
31 …_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
32 …output_weights = Input("input_to_output_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
Dlstm_state.mod.py22 n_input = 2 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
29 …o_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
30 …forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
31 …_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
32 …output_weights = Input("input_to_output_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
Dlstm3_state2.mod.py22 n_input = 5 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
29 …o_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
30 …forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
31 …_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
32 …output_weights = Input("input_to_output_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
Dlstm3_state.mod.py22 n_input = 5 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
29 …o_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
30 …forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
31 …_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
32 …output_weights = Input("input_to_output_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
/packages/modules/NeuralNetworks/tools/test_generator/tests/P_vts_backward_compatibility_float/
Dlstm_float.mod.py18 n_input = 2 variable
23 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
25 …o_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
26 …forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
27 …_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))
28 …output_weights = Input("input_to_output_weights", "TENSOR_FLOAT32", "{%d, %d}" % (n_cell, n_input))

123