Home
last modified time | relevance | path

Searched refs:n_batch (Results 1 – 25 of 54) sorted by relevance

123

/frameworks/ml/nn/runtime/test/specs/V1_0/
Dlstm3_state3.mod.py21 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 …]
Dlstm2_state2.mod.py21 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 …]
Dlstm3.mod.py21 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 …]
Dlstm_state2.mod.py21 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 …]
Dlstm2.mod.py21 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 …]
Dlstm2_state.mod.py21 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) ],
Dlstm_state.mod.py21 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) ],
Dlstm3_state2.mod.py21 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))
644 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
Dlstm.mod.py21 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_1/
Dlstm2_state2_relaxed.mod.py21 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 …]
Dlstm2_relaxed.mod.py21 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 …]
Dlstm_state2_relaxed.mod.py21 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 …]
Dlstm3_relaxed.mod.py21 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))
622 input0[cell_state_in] = [ 0 for _ in range(n_batch * n_cell) ]
623 input0[output_state_in] = [ 0 for _ in range(n_batch * n_output) ]
[all …]
Dlstm3_state3_relaxed.mod.py21 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))
645 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
646 cell_state_out: [ 0 for x in range(n_batch * n_cell) ],
[all …]
/frameworks/ml/nn/runtime/test/specs/V1_2/
Dlstm3_state3_float16.mod.py21 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))
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 …]
Dlstm2_state2_float16.mod.py21 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 …]
Dlstm_state2_float16.mod.py21 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 …]
Dlstm3_float16.mod.py21 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))
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 …]
Dlstm2_float16.mod.py21 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 …]
Dlayer_norm_lstm.mod.py22 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 …]
Dquantized_lstm.mod.py20 n_batch = 2 variable
25 input_ = Input("input", ("TENSOR_QUANT8_ASYMM", (n_batch, n_input), 1 / 128, 128))
45 prev_cell_state = Input("prevCellState", ("TENSOR_QUANT16_SYMM", (n_batch, n_cell), 1 / 2048, 0))
46 prev_output = Input("prevOutput", ("TENSOR_QUANT8_ASYMM", (n_batch, n_output), 1 / 128, 128))
48 cell_state_out = Output("cellStateOut", ("TENSOR_QUANT16_SYMM", (n_batch, n_cell), 1 / 2048, 0))
49 output = Output("output", ("TENSOR_QUANT8_ASYMM", (n_batch, n_output), 1 / 128, 128))
99 n_batch = 1 variable
105 ("TENSOR_QUANT8_ASYMM", (n_batch, n_input), 1 / 128, 128))
170 ("TENSOR_QUANT16_SYMM", (n_batch, n_cell), 1 / 2048, 0))
172 ("TENSOR_QUANT8_ASYMM", (n_batch, n_output), 1 / 128, 128))
[all …]
Dlstm2_state_float16.mod.py21 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) ],
Dlstm_state_float16.mod.py21 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/common/operations/
DLSTM.cpp305 const uint32_t n_batch = SizeOfDimension(input_, 0); in Prepare() local
333 outputShape->dimensions = {n_batch, n_output}; in Prepare()
338 outputStateShape->dimensions = {n_batch, n_output}; in Prepare()
343 cellStateShape->dimensions = {n_batch, n_cell}; in Prepare()
349 scratchShape->dimensions = {n_batch, n_cell * 3}; in Prepare()
352 scratchShape->dimensions = {n_batch, n_cell * 4}; in Prepare()
745 const uint32_t n_batch = input_shape.dimensions[0]; in LSTMStep() local
759 forget_gate_scratch = cell_scratch + n_cell * n_batch; in LSTMStep()
760 output_gate_scratch = cell_scratch + 2 * n_cell * n_batch; in LSTMStep()
763 cell_scratch = input_gate_scratch + n_cell * n_batch; in LSTMStep()
[all …]
/frameworks/ml/nn/tools/test_generator/tests/P_vts_backward_compatibility_float/
Dlstm_float.mod.py17 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) ],

123