| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | layer_norm_lstm.mod.py | 27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
29 input_to_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT32",
31 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32",
33 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT32",
35 input_to_output_weights = Input("input_to_output_weights", "TENSOR_FLOAT32",
38 recurrent_to_input_weights = Input("recurrent_to_intput_weights",
41 recurrent_to_forget_weights = Input("recurrent_to_forget_weights",
44 recurrent_to_cell_weights = Input("recurrent_to_cell_weights", "TENSOR_FLOAT32",
46 recurrent_to_output_weights = Input("recurrent_to_output_weights",
50 cell_to_input_weights = Input("cell_to_input_weights", "TENSOR_FLOAT32",
[all …]
|
| D | bidirectional_sequence_rnn.mod.py | 209 input=Input("input", "TENSOR_FLOAT32", "{{ {}, {}, {} }}".format(
211 fw_weights=Input("fw_weights", "TENSOR_FLOAT32", "{{ {}, {} }}".format(
213 fw_recurrent_weights=Input(
216 fw_bias=Input("fw_bias", "TENSOR_FLOAT32", "{{ {} }}".format(fw_num_units)),
217 fw_hidden_state=Input("fw_hidden_state", "TENSOR_FLOAT32",
219 bw_weights=Input("bw_weights", "TENSOR_FLOAT32", "{{ {}, {} }}".format(
221 bw_recurrent_weights=Input(
224 bw_bias=Input("bw_bias", "TENSOR_FLOAT32", "{{ {} }}".format(bw_num_units)),
225 bw_hidden_state=Input("bw_hidden_state", "TENSOR_FLOAT32",
227 aux_input=Input("aux_input", "TENSOR_FLOAT32", "{0}"),
[all …]
|
| D | bidirectional_sequence_lstm_merge_outputs.mod.py | 27 input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input))
29 fw_input_to_input_weights = Input(
31 fw_input_to_forget_weights = Input(
33 fw_input_to_cell_weights = Input(
35 fw_input_to_output_weights = Input(
38 fw_recurrent_to_input_weights = Input(
40 fw_recurrent_to_forget_weights = Input(
42 fw_recurrent_to_cell_weights = Input(
44 fw_recurrent_to_output_weights = Input(
47 fw_cell_to_input_weights = Input(
[all …]
|
| D | bidirectional_sequence_lstm_float16_batch_major_merge_outputs.mod.py | 27 input = Input("input", "TENSOR_FLOAT16", "{{{}, {}, {}}}".format(n_batch, max_time, n_input))
29 fw_input_to_input_weights = Input(
31 fw_input_to_forget_weights = Input(
33 fw_input_to_cell_weights = Input(
35 fw_input_to_output_weights = Input(
38 fw_recurrent_to_input_weights = Input(
40 fw_recurrent_to_forget_weights = Input(
42 fw_recurrent_to_cell_weights = Input(
44 fw_recurrent_to_output_weights = Input(
47 fw_cell_to_input_weights = Input(
[all …]
|
| D | bidirectional_sequence_lstm_cifg_peephole.mod.py | 26 input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input))
28 fw_input_to_input_weights = Input(
30 fw_input_to_forget_weights = Input(
32 fw_input_to_cell_weights = Input(
34 fw_input_to_output_weights = Input(
37 fw_recurrent_to_input_weights = Input(
39 fw_recurrent_to_forget_weights = Input(
41 fw_recurrent_to_cell_weights = Input(
43 fw_recurrent_to_output_weights = Input(
46 fw_cell_to_input_weights = Input(
[all …]
|
| D | bidirectional_sequence_lstm_float16_batch_major.mod.py | 26 input = Input("input", "TENSOR_FLOAT16", "{{{}, {}, {}}}".format(n_batch, max_time, n_input))
28 fw_input_to_input_weights = Input(
30 fw_input_to_forget_weights = Input(
32 fw_input_to_cell_weights = Input(
34 fw_input_to_output_weights = Input(
37 fw_recurrent_to_input_weights = Input(
39 fw_recurrent_to_forget_weights = Input(
41 fw_recurrent_to_cell_weights = Input(
43 fw_recurrent_to_output_weights = Input(
46 fw_cell_to_input_weights = Input(
[all …]
|
| D | bidirectional_sequence_lstm.mod.py | 26 input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input))
28 fw_input_to_input_weights = Input(
30 fw_input_to_forget_weights = Input(
32 fw_input_to_cell_weights = Input(
34 fw_input_to_output_weights = Input(
37 fw_recurrent_to_input_weights = Input(
39 fw_recurrent_to_forget_weights = Input(
41 fw_recurrent_to_cell_weights = Input(
43 fw_recurrent_to_output_weights = Input(
46 fw_cell_to_input_weights = Input(
[all …]
|
| D | unidirectional_sequence_lstm_layer_norm_cifg_peephole.mod.py | 30 input = Input("input", "TENSOR_FLOAT32", "{%d, %d, %d}" % (max_time, n_batch, n_input))
32 input_to_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT32",
34 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32",
36 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT32",
38 input_to_output_weights = Input("input_to_output_weights", "TENSOR_FLOAT32",
41 recurrent_to_input_weights = Input("recurrent_to_intput_weights",
44 recurrent_to_forget_weights = Input("recurrent_to_forget_weights",
47 recurrent_to_cell_weights = Input("recurrent_to_cell_weights", "TENSOR_FLOAT32",
49 recurrent_to_output_weights = Input("recurrent_to_output_weights",
53 cell_to_input_weights = Input("cell_to_input_weights", "TENSOR_FLOAT32",
[all …]
|
| D | unidirectional_sequence_lstm_batch_major_peephole_projection_bias.mod.py | 30 input = Input("input", "TENSOR_FLOAT32", "{%d, %d, %d}" % (n_batch, max_time, n_input))
32 input_to_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT32",
34 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32",
36 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT32",
38 input_to_output_weights = Input("input_to_output_weights", "TENSOR_FLOAT32",
41 recurrent_to_input_weights = Input("recurrent_to_intput_weights",
44 recurrent_to_forget_weights = Input("recurrent_to_forget_weights",
47 recurrent_to_cell_weights = Input("recurrent_to_cell_weights", "TENSOR_FLOAT32",
49 recurrent_to_output_weights = Input("recurrent_to_output_weights",
53 cell_to_input_weights = Input("cell_to_input_weights", "TENSOR_FLOAT32",
[all …]
|
| D | bidirectional_sequence_lstm_aux_input.mod.py | 28 input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input))
30 fw_input_to_input_weights = Input(
32 fw_input_to_forget_weights = Input(
34 fw_input_to_cell_weights = Input(
36 fw_input_to_output_weights = Input(
39 fw_recurrent_to_input_weights = Input(
41 fw_recurrent_to_forget_weights = Input(
43 fw_recurrent_to_cell_weights = Input(
45 fw_recurrent_to_output_weights = Input(
48 fw_cell_to_input_weights = Input(
[all …]
|
| D | bidirectional_sequence_lstm_norm_fw_output.mod.py | 27 input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input))
29 fw_input_to_input_weights = Input(
31 fw_input_to_forget_weights = Input(
33 fw_input_to_cell_weights = Input(
35 fw_input_to_output_weights = Input(
38 fw_recurrent_to_input_weights = Input(
40 fw_recurrent_to_forget_weights = Input(
42 fw_recurrent_to_cell_weights = Input(
44 fw_recurrent_to_output_weights = Input(
47 fw_cell_to_input_weights = Input(
[all …]
|
| D | bidirectional_sequence_lstm_float16_batch_major_aux_input.mod.py | 29 input = Input("input", "TENSOR_FLOAT16", "{{{}, {}, {}}}".format(n_batch, max_time, n_input))
31 fw_input_to_input_weights = Input(
33 fw_input_to_forget_weights = Input(
35 fw_input_to_cell_weights = Input(
37 fw_input_to_output_weights = Input(
40 fw_recurrent_to_input_weights = Input(
42 fw_recurrent_to_forget_weights = Input(
44 fw_recurrent_to_cell_weights = Input(
46 fw_recurrent_to_output_weights = Input(
49 fw_cell_to_input_weights = Input(
[all …]
|
| D | slice.mod.py | 25 inp=Input("input", "TENSOR_FLOAT32", "{4}"),
27 begin=Input("begin", "TENSOR_INT32", "{1}"),
29 size=Input("size", "TENSOR_INT32", "{1}"),
34 inp=Input("input", "TENSOR_FLOAT32", "{2,3}"),
36 begin=Input("begin", "TENSOR_INT32", "{2}"),
38 size=Input("size", "TENSOR_INT32", "{2}"),
43 inp=Input("input", "TENSOR_FLOAT32", "{2,3,2}"),
45 begin=Input("begin", "TENSOR_INT32", "{3}"),
47 size=Input("size", "TENSOR_INT32", "{3}"),
52 inp=Input("input", "TENSOR_FLOAT32", "{4, 1, 1, 1}"),
[all …]
|
| D | unidirectional_sequence_lstm_f16_batch_major.mod.py | 31 input = Input("input", "TENSOR_FLOAT16", "{%d, %d, %d}" % (n_batch, max_time, n_input))
33 input_to_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT16",
35 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT16",
37 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT16",
39 input_to_output_weights = Input("input_to_output_weights", "TENSOR_FLOAT16",
42 recurrent_to_input_weights = Input("recurrent_to_intput_weights",
45 recurrent_to_forget_weights = Input("recurrent_to_forget_weights",
48 recurrent_to_cell_weights = Input("recurrent_to_cell_weights", "TENSOR_FLOAT16",
50 recurrent_to_output_weights = Input("recurrent_to_output_weights",
54 cell_to_input_weights = Input("cell_to_input_weights", "TENSOR_FLOAT16",
[all …]
|
| D | unidirectional_sequence_lstm_batch_major_norm_peephole_projection.mod.py | 30 input = Input("input", "TENSOR_FLOAT32", "{%d, %d, %d}" % (n_batch, max_time, n_input))
32 input_to_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT32",
34 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32",
36 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT32",
38 input_to_output_weights = Input("input_to_output_weights", "TENSOR_FLOAT32",
41 recurrent_to_input_weights = Input("recurrent_to_intput_weights",
44 recurrent_to_forget_weights = Input("recurrent_to_forget_weights",
47 recurrent_to_cell_weights = Input("recurrent_to_cell_weights", "TENSOR_FLOAT32",
49 recurrent_to_output_weights = Input("recurrent_to_output_weights",
53 cell_to_input_weights = Input("cell_to_input_weights", "TENSOR_FLOAT32",
[all …]
|
| D | unidirectional_sequence_lstm_cifg_peephole.mod.py | 30 input = Input("input", "TENSOR_FLOAT32", "{%d, %d, %d}" % (max_time, n_batch, n_input))
32 input_to_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT32",
34 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32",
36 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT32",
38 input_to_output_weights = Input("input_to_output_weights", "TENSOR_FLOAT32",
41 recurrent_to_input_weights = Input("recurrent_to_intput_weights",
44 recurrent_to_forget_weights = Input("recurrent_to_forget_weights",
47 recurrent_to_cell_weights = Input("recurrent_to_cell_weights", "TENSOR_FLOAT32",
49 recurrent_to_output_weights = Input("recurrent_to_output_weights",
53 cell_to_input_weights = Input("cell_to_input_weights", "TENSOR_FLOAT32",
[all …]
|
| D | unidirectional_sequence_lstm_norm_peephole_projection.mod.py | 30 input = Input("input", "TENSOR_FLOAT32", "{%d, %d, %d}" % (max_time, n_batch, n_input))
32 input_to_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT32",
34 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32",
36 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT32",
38 input_to_output_weights = Input("input_to_output_weights", "TENSOR_FLOAT32",
41 recurrent_to_input_weights = Input("recurrent_to_intput_weights",
44 recurrent_to_forget_weights = Input("recurrent_to_forget_weights",
47 recurrent_to_cell_weights = Input("recurrent_to_cell_weights", "TENSOR_FLOAT32",
49 recurrent_to_output_weights = Input("recurrent_to_output_weights",
53 cell_to_input_weights = Input("cell_to_input_weights", "TENSOR_FLOAT32",
[all …]
|
| D | unidirectional_sequence_lstm_1step.mod.py | 30 input = Input("input", "TENSOR_FLOAT32", "{%d, %d, %d}" % (max_time, n_batch, n_input))
32 input_to_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT32",
34 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32",
36 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT32",
38 input_to_output_weights = Input("input_to_output_weights", "TENSOR_FLOAT32",
41 recurrent_to_input_weights = Input("recurrent_to_intput_weights",
44 recurrent_to_forget_weights = Input("recurrent_to_forget_weights",
47 recurrent_to_cell_weights = Input("recurrent_to_cell_weights", "TENSOR_FLOAT32",
49 recurrent_to_output_weights = Input("recurrent_to_output_weights",
53 cell_to_input_weights = Input("cell_to_input_weights", "TENSOR_FLOAT32",
[all …]
|
| D | unidirectional_sequence_lstm_f16_norm_peephole_projection.mod.py | 30 input = Input("input", "TENSOR_FLOAT16", "{%d, %d, %d}" % (max_time, n_batch, n_input))
32 input_to_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT16",
34 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT16",
36 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT16",
38 input_to_output_weights = Input("input_to_output_weights", "TENSOR_FLOAT16",
41 recurrent_to_input_weights = Input("recurrent_to_intput_weights",
44 recurrent_to_forget_weights = Input("recurrent_to_forget_weights",
47 recurrent_to_cell_weights = Input("recurrent_to_cell_weights", "TENSOR_FLOAT16",
49 recurrent_to_output_weights = Input("recurrent_to_output_weights",
53 cell_to_input_weights = Input("cell_to_input_weights", "TENSOR_FLOAT16",
[all …]
|
| D | lstm2_state_float16.mod.py | 27 input = Input("input", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_input))
29 input_to_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_…
30 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, …
31 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, n_in…
32 input_to_output_weights = Input("input_to_output_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_cell, …
34 recurrent_to_input_weights = Input("recurrent_to_intput_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n…
35 recurrent_to_forget_weights = Input("recurrent_to_forget_weights", "TENSOR_FLOAT16", "{%d, %d}" % (…
36 recurrent_to_cell_weights = Input("recurrent_to_cell_weights", "TENSOR_FLOAT16", "{%d, %d}" % (n_ce…
37 recurrent_to_output_weights = Input("recurrent_to_output_weights", "TENSOR_FLOAT16", "{%d, %d}" % (…
39 cell_to_input_weights = Input("cell_to_input_weights", "TENSOR_FLOAT16", "{0}")
[all …]
|
| /frameworks/ml/nn/runtime/test/specs/V1_3/ |
| D | bidirectional_sequence_lstm_state_output.mod.py | 26 input = Input("input", "TENSOR_FLOAT32", "{{{}, {}, {}}}".format(max_time, n_batch, n_input))
28 fw_input_to_input_weights = Input(
30 fw_input_to_forget_weights = Input(
32 fw_input_to_cell_weights = Input(
34 fw_input_to_output_weights = Input(
37 fw_recurrent_to_input_weights = Input(
39 fw_recurrent_to_forget_weights = Input(
41 fw_recurrent_to_cell_weights = Input(
43 fw_recurrent_to_output_weights = Input(
46 fw_cell_to_input_weights = Input(
[all …]
|
| D | bidirectional_sequence_lstm.mod.py | 87 input = Input("input", "TENSOR_FLOAT32",
90 fw_input_to_input_weights = Input("fw_input_to_input_weights",
93 fw_input_to_forget_weights = Input("fw_input_to_forget_weights",
96 fw_input_to_cell_weights = Input("fw_input_to_cell_weights", "TENSOR_FLOAT32",
98 fw_input_to_output_weights = Input("fw_input_to_output_weights",
102 fw_recurrent_to_input_weights = Input("fw_recurrent_to_input_weights",
105 fw_recurrent_to_forget_weights = Input("fw_recurrent_to_forget_weights",
108 fw_recurrent_to_cell_weights = Input("fw_recurrent_to_cell_weights",
111 fw_recurrent_to_output_weights = Input("fw_recurrent_to_output_weights",
115 fw_cell_to_input_weights = Input("fw_cell_to_input_weights", "TENSOR_FLOAT32",
[all …]
|
| D | bidirectional_sequence_rnn_state_output.mod.py | 220 input=Input("input", "TENSOR_FLOAT32",
222 fw_weights=Input("fw_weights", "TENSOR_FLOAT32",
224 fw_recurrent_weights=Input(
227 fw_bias=Input("fw_bias", "TENSOR_FLOAT32", "{{ {} }}".format(fw_num_units)),
228 fw_hidden_state=Input("fw_hidden_state", "TENSOR_FLOAT32",
230 bw_weights=Input("bw_weights", "TENSOR_FLOAT32",
232 bw_recurrent_weights=Input(
235 bw_bias=Input("bw_bias", "TENSOR_FLOAT32", "{{ {} }}".format(bw_num_units)),
236 bw_hidden_state=Input("bw_hidden_state", "TENSOR_FLOAT32",
238 aux_input=Input("aux_input", "TENSOR_FLOAT32", "{0}"),
[all …]
|
| D | unidirectional_sequence_lstm_layer_norm_cifg_peephole_state_output.mod.py | 30 input = Input("input", "TENSOR_FLOAT32",
33 input_to_input_weights = Input("input_to_input_weights", "TENSOR_FLOAT32",
35 input_to_forget_weights = Input("input_to_forget_weights", "TENSOR_FLOAT32",
37 input_to_cell_weights = Input("input_to_cell_weights", "TENSOR_FLOAT32",
39 input_to_output_weights = Input("input_to_output_weights", "TENSOR_FLOAT32",
42 recurrent_to_input_weights = Input("recurrent_to_intput_weights",
44 recurrent_to_forget_weights = Input("recurrent_to_forget_weights",
47 recurrent_to_cell_weights = Input("recurrent_to_cell_weights", "TENSOR_FLOAT32",
49 recurrent_to_output_weights = Input("recurrent_to_output_weights",
53 cell_to_input_weights = Input("cell_to_input_weights", "TENSOR_FLOAT32", "{0}")
[all …]
|
| D | qlstm_noprojection.mod.py | 28 input = Input("input", InputType)
31 input_to_input_weights = Input("input_to_input_weights", InputWeightsType)
32 input_to_forget_weights = Input("input_to_forget_weights", InputWeightsType)
33 input_to_cell_weights = Input("input_to_cell_weights", InputWeightsType)
34 input_to_output_weights = Input("input_to_output_weights", InputWeightsType)
37 recurrent_to_input_weights = Input("recurrent_to_input_weights", RecurrentWeightsType)
38 recurrent_to_forget_weights = Input("recurrent_to_forget_weights", RecurrentWeightsType)
39 recurrent_to_cell_weights = Input("recurrent_to_cell_weights", RecurrentWeightsType)
40 recurrent_to_output_weights = Input("recurrent_to_output_weights", RecurrentWeightsType)
43 cell_to_input_weights = Input("cell_to_input_weights", CellWeightsType)
[all …]
|