| /frameworks/ml/nn/runtime/test/specs/V1_1/ |
| 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 * 4)))
59 output_state_out = IgnoredOutput("output_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_output))
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))
[all...] |
| lstm3_state2_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 * 4)))
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))
[all...] |
| lstm3_state_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 * 4)))
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))
[all...] |
| /frameworks/ml/nn/common/operations/ |
| LSTMTest.cpp | 76 LSTMOpModel(uint32_t n_batch, uint32_t n_input,
91 input_shapes.push_back({n_batch, n_output});
92 input_shapes.push_back({n_batch, n_cell});
114 {n_batch, n_cell * (use_cifg ? 3 : 4)},
115 {n_batch, n_output},
116 {n_batch, n_cell},
117 {n_batch, n_output},
134 Input_.insert(Input_.end(), n_batch * n_input, 0.f);
135 OutputStateIn_.insert(OutputStateIn_.end(), n_batch * n_output, 0.f);
136 CellStateIn_.insert(CellStateIn_.end(), n_batch * n_cell, 0.f)
274 const int n_batch = 1; local
385 const int n_batch = 1; local
490 const int n_batch = 2; local
[all...] |
| LSTM.h | 61 // Input Tensors of size {n_batch, n_input}
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| neon_tensor_utils.cc | 32 int n_batch, float* result,
45 for (int b = 0; b < n_batch; b++) {
163 int n_batch, float* result) {
180 for (int b = 0; b < n_batch; b++) {
278 int n_batch, float* result,
283 for (int b = 0; b < n_batch; b++) {
|
| /frameworks/ml/nn/runtime/test/specs/V1_0/ |
| 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 * 4)))
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))
[all...] |
| lstm3_state2.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 * 4)))
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))
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| lstm_test.cc | 35 LSTMOpModel(int n_batch, int n_input, int n_cell, int n_output, bool use_cifg,
39 : n_batch_(n_batch),
245 const int n_batch = 1; local
251 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output,
257 {n_batch, n_input}, // input tensor
355 const int n_batch = 1; local
361 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output,
367 {n_batch, n_input}, // input tensor
460 const int n_batch = 2; local
465 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output
[all...] |
| unidirectional_sequence_lstm_test.cc | 35 UnidirectionalLSTMOpModel(int n_batch, int n_input, int n_cell, int n_output,
41 : n_batch_(n_batch),
251 const int n_batch = 1; local
259 n_batch, n_input, n_cell, n_output, sequence_length, /*use_cifg=*/false,
263 {sequence_length, n_batch, n_input}, // input tensor
359 const int n_batch = 1; local
367 n_batch, n_input, n_cell, n_output, sequence_length, /*use_cifg=*/true,
371 {sequence_length, n_batch, n_input}, // input tensor
461 const int n_batch = 2; local
468 n_batch, n_input, n_cell, n_output, sequence_length, /*use_cifg=*/false
[all...] |
| /external/deqp/external/openglcts/modules/gl/ |
| gl4cVertexAttrib64BitTest.cpp | [all...] |
