| /frameworks/ml/nn/runtime/test/specs/V1_0/ |
| D | lstm_state2.mod.py | 21 n_batch = 1 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, (n_cell *…
59 output_state_out = IgnoredOutput("output_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_out…
60 cell_state_out = IgnoredOutput("cell_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
140 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
141 cell_state_out: [ 0 for x in range(n_batch * n_cell) ],
[all …]
|
| D | lstm2.mod.py | 21 n_batch = 1 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell * …
59 output_state_out = Output("output_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
60 cell_state_out = Output("cell_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
132 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 3) ],
138 input0[output_state_in] = [ 0 for _ in range(n_batch * n_output) ]
[all …]
|
| D | lstm2_state2.mod.py | 21 n_batch = 1 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell * …
59 output_state_out = IgnoredOutput("output_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_out…
60 cell_state_out = IgnoredOutput("cell_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
132 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 3) ],
133 cell_state_out: [ 0 for x in range(n_batch * n_cell) ],
[all …]
|
| D | lstm3.mod.py | 21 n_batch = 2 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, (n_cell *…
59 output_state_out = Output("output_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
60 cell_state_out = Output("cell_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
620 input0[cell_state_in] = [ 0 for _ in range(n_batch * n_cell) ]
621 input0[output_state_in] = [ 0 for _ in range(n_batch * n_output) ]
[all …]
|
| D | lstm3_state3.mod.py | 21 n_batch = 2 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, (n_cell *…
59 output_state_out = IgnoredOutput("output_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_out…
60 cell_state_out = IgnoredOutput("cell_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
643 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
644 cell_state_out: [ 0 for x in range(n_batch * n_cell) ],
[all …]
|
| D | lstm2_state.mod.py | 21 n_batch = 1 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell * …
59 output_state_out = Output("output_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
60 cell_state_out = Output("cell_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
132 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 3) ],
|
| D | lstm3_state.mod.py | 21 n_batch = 2 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, (n_cell *…
59 output_state_out = Output("output_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
60 cell_state_out = Output("cell_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
643 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
|
| D | lstm.mod.py | 21 n_batch = 1 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, (n_cell *…
59 output_state_out = Output("output_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
60 cell_state_out = Output("cell_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
140 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
|
| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | lstm2_state2_float16.mod.py | 21 n_batch = 1 variable
27 input = Input("input", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_cell * …
59 output_state_out = IgnoredOutput("output_state_out", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_out…
60 cell_state_out = IgnoredOutput("cell_state_out", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_output))
132 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 3) ],
133 cell_state_out: [ 0 for x in range(n_batch * n_cell) ],
[all …]
|
| D | lstm2_float16.mod.py | 21 n_batch = 1 variable
27 input = Input("input", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_cell * …
59 output_state_out = Output("output_state_out", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_output))
60 cell_state_out = Output("cell_state_out", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_output))
132 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 3) ],
138 input0[output_state_in] = [ 0 for _ in range(n_batch * n_output) ]
[all …]
|
| D | lstm_state2_float16.mod.py | 21 n_batch = 1 variable
27 input = Input("input", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, (n_cell *…
59 output_state_out = IgnoredOutput("output_state_out", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_out…
60 cell_state_out = IgnoredOutput("cell_state_out", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_output))
140 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
141 cell_state_out: [ 0 for x in range(n_batch * n_cell) ],
[all …]
|
| D | lstm3_float16.mod.py | 21 n_batch = 2 variable
27 input = Input("input", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, (n_cell *…
59 output_state_out = Output("output_state_out", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_output))
60 cell_state_out = Output("cell_state_out", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_output))
620 input0[cell_state_in] = [ 0 for _ in range(n_batch * n_cell) ]
621 input0[output_state_in] = [ 0 for _ in range(n_batch * n_output) ]
[all …]
|
| D | lstm3_state3_float16.mod.py | 21 n_batch = 2 variable
27 input = Input("input", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, (n_cell *…
59 output_state_out = IgnoredOutput("output_state_out", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_out…
60 cell_state_out = IgnoredOutput("cell_state_out", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_output))
643 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
644 cell_state_out: [ 0 for x in range(n_batch * n_cell) ],
[all …]
|
| D | layer_norm_lstm.mod.py | 22 n_batch = 2 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
69 "{%d, %d}" % (n_batch, n_output))
71 "{%d, %d}" % (n_batch, n_cell))
87 "{%d, %d}" % (n_batch, (n_cell * 4)))
89 "{%d, %d}" % (n_batch, n_output))
91 "{%d, %d}" % (n_batch, n_cell))
92 output = Output("output", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
183 scratch_buffer: [0] * (n_batch * n_cell * 4),
194 n_batch = 2 variable
[all …]
|
| D | lstm2_state_float16.mod.py | 21 n_batch = 1 variable
27 input = Input("input", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_cell * …
59 output_state_out = Output("output_state_out", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_output))
60 cell_state_out = Output("cell_state_out", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_output))
132 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 3) ],
|
| D | lstm_state_float16.mod.py | 21 n_batch = 1 variable
27 input = Input("input", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, (n_cell *…
59 output_state_out = Output("output_state_out", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_output))
60 cell_state_out = Output("cell_state_out", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT16", "{%d, %d}" % (n_batch, n_output))
140 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
|
| /frameworks/ml/nn/runtime/test/specs/V1_1/ |
| D | lstm2_state2_relaxed.mod.py | 21 n_batch = 1 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell * …
59 output_state_out = IgnoredOutput("output_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_out…
60 cell_state_out = IgnoredOutput("cell_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
133 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 3) ],
134 cell_state_out: [ 0 for x in range(n_batch * n_cell) ],
[all …]
|
| D | lstm_state2_relaxed.mod.py | 21 n_batch = 1 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, (n_cell *…
59 output_state_out = IgnoredOutput("output_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_out…
60 cell_state_out = IgnoredOutput("cell_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
141 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
142 cell_state_out: [ 0 for x in range(n_batch * n_cell) ],
[all …]
|
| D | lstm2_relaxed.mod.py | 21 n_batch = 1 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell * …
59 output_state_out = Output("output_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
60 cell_state_out = Output("cell_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
133 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 3) ],
139 input0[output_state_in] = [ 0 for _ in range(n_batch * n_output) ]
[all …]
|
| D | lstm3_state3_relaxed.mod.py | 21 n_batch = 2 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, (n_cell *…
59 output_state_out = IgnoredOutput("output_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_out…
60 cell_state_out = IgnoredOutput("cell_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
644 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
645 cell_state_out: [ 0 for x in range(n_batch * n_cell) ],
[all …]
|
| D | lstm3_relaxed.mod.py | 21 n_batch = 2 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, (n_cell *…
59 output_state_out = Output("output_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
60 cell_state_out = Output("cell_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
621 input0[cell_state_in] = [ 0 for _ in range(n_batch * n_cell) ]
622 input0[output_state_in] = [ 0 for _ in range(n_batch * n_output) ]
[all …]
|
| D | lstm2_state_relaxed.mod.py | 21 n_batch = 1 variable
27 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
51 output_state_in = Input("output_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
52 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell * …
59 output_state_out = Output("output_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
60 cell_state_out = Output("cell_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
61 output = Output("output", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
133 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 3) ],
|
| /frameworks/ml/nn/runtime/test/specs/V1_3/ |
| D | bidirectional_sequence_lstm.mod.py | 22 n_batch, argument
88 "{{{}, {}, {}}}".format(max_time, n_batch, n_fw_input))
181 "{{{}, {}}}".format(n_batch, n_output))
183 "{{{}, {}}}".format(n_batch, n_cell))
186 "{{{}, {}}}".format(n_batch, n_output))
188 "{{{}, {}}}".format(n_batch, n_cell))
191 "{{{}, {}, {}}}".format(max_time, n_batch, n_bw_input))
230 "{{{}, {}, {}}}".format(max_time, n_batch, n_output))
232 "{{{}, {}, {}}}".format(max_time, n_batch, n_output))
237 "{{{}, {}, {}}}".format(max_time, n_batch, n_output))
[all …]
|
| D | unidirectional_sequence_lstm_layer_norm_cifg_peephole_state_output.mod.py | 24 n_batch = 2 variable
31 "{%d, %d, %d}" % (max_time, n_batch, n_input))
71 "{%d, %d}" % (n_batch, n_output))
73 "{%d, %d}" % (n_batch, n_cell))
90 "{%d, %d, %d}" % (max_time, n_batch, n_output))
92 "{%d, %d}" % (n_batch, n_output))
94 "{%d, %d}" % (n_batch, n_cell))
181 golden_output[(max_time - 1) * (n_batch * n_output):],
189 input0[output_state_in] = [0 for _ in range(n_batch * n_output)]
190 input0[cell_state_in] = [0 for _ in range(n_batch * n_cell)]
|
| /frameworks/ml/nn/tools/test_generator/tests/P_backward_compatibility_float/ |
| D | lstm_float.mod.py | 17 n_batch = 1 variable
23 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_input))
47 output_state_in = Input("output_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
48 cell_state_in = Input("cell_state_in", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
54 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, (n_cell *…
55 output_state_out = Output("output_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
56 cell_state_out = Output("cell_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell))
57 output = Output("output", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
137 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
|