Home
last modified time | relevance | path

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

123

/aosp12/packages/modules/NeuralNetworks/common/
H A DQuantUtils.cpp17 for (int j = 0; j < n_input; ++j) { in ApplyLayerNorm()
18 const int32_t index = i * n_input + j; 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()
88 for (int c = 0; c < n_input; c++) { in ApplySigmoid()
102 for (int i = 0; i < n_input; ++i) { in CwiseMul()
115 for (int i = 0; i < n_input; ++i) { in CwiseMul()
141 for (int i = 0; i < n_input; ++i) { in CwiseAdd()
152 for (int i = 0; i < n_input; ++i) { in CwiseClipping()
[all …]
H A DQuantUtils.h32 int32_t shift, int32_t n_batch, int32_t n_input, in MatrixBatchVectorMultiplyAccumulate() argument
39 for (int col = 0; col < n_input; ++col) { in MatrixBatchVectorMultiplyAccumulate()
40 int8_t input_val = input[batch * n_input + col]; in MatrixBatchVectorMultiplyAccumulate()
41 int8_t weights_val = input_to_gate_weights[row * n_input + col]; in MatrixBatchVectorMultiplyAccumulate()
138 int n_batch, int n_input, int16_t* output);
150 void ApplyTanh(const int16_t* input, int32_t n_batch, int32_t n_input, int16_t* output) { in ApplyTanh() argument
154 for (int i = 0; i < n_input; ++i) { in ApplyTanh()
155 const int index = batch * n_input + i; in ApplyTanh()
168 ApplyTanh<i>(input, n_batch, n_input, output); \ in ApplyTanh()
187 int32_t n_batch, int32_t n_input, int32_t output_zp, int8_t* output);
[all …]
/aosp12/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/
H A 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 …]
H A 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 …]
H A 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 …]
H A 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 …]
H A 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 …]
H A 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 …]
H A 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 …]
H A 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 …]
H A 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 …]
H A 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))
/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DLSTMTest.cpp82 : 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()
378 const int n_input = 2; in TEST() local
383 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST()
389 {n_batch, n_input}, // input tensor in TEST()
475 const int n_input = 5; in TEST() local
479 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST()
[all …]
H A DLayerNormLSTMTest.cpp90 : n_input_(n_input), in LayerNormLSTMOpModel()
147 Input_.insert(Input_.end(), n_batch * n_input, 0.f); in LayerNormLSTMOpModel()
287 const int n_input = 5; in TEST() local
292 LayerNormLSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST()
298 {n_batch, n_input}, // input tensor in TEST()
300 {n_cell, n_input}, // input_to_input_weight tensor in TEST()
302 {n_cell, n_input}, // input_to_cell_weight tensor in TEST()
404 const int input_sequence_size = lstm_input[0].size() / n_input; in TEST()
407 const float* batch_start = lstm_input[b].data() + i * n_input; in TEST()
408 const float* batch_end = batch_start + n_input; in TEST()
[all …]
/aosp12/packages/modules/NeuralNetworks/runtime/test/specs/V1_3/
H A 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 …]
/aosp12/packages/modules/NeuralNetworks/runtime/test/specs/V1_0/
H A 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))
H A Dlstm.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))
H A Dlstm2.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))
H A 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))
H A Dlstm3.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))
H A 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))
H A 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))
H A Dlstm3_state3.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))
H A 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))
H A Dlstm_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))

123