| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | layer_norm_lstm.mod.py | 86 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", variable
104 [scratch_buffer, output_state_out, cell_state_out, output])
183 scratch_buffer: [0] * (n_batch * n_cell * 4),
258 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", variable
276 [scratch_buffer, output_state_out, cell_state_out, output])
352 scratch_buffer: [0] * (n_batch * n_cell * 3),
|
| /frameworks/ml/nn/runtime/test/generated/models/ |
| D | lstm_state.model.cpp | 38 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
49 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
53 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
96 auto scratch_buffer = model->addOperand(&type5); in CreateModel_dynamic_output_shape() local
107 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
111 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm_state2.model.cpp | 38 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
49 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
53 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
96 auto scratch_buffer = model->addOperand(&type5); in CreateModel_dynamic_output_shape() local
107 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
111 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm.model.cpp | 38 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
49 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
53 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
96 auto scratch_buffer = model->addOperand(&type5); in CreateModel_dynamic_output_shape() local
107 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
111 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm2_state.model.cpp | 38 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
49 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
53 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
96 auto scratch_buffer = model->addOperand(&type5); in CreateModel_dynamic_output_shape() local
107 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
111 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm2_state2.model.cpp | 38 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
49 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
53 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
96 auto scratch_buffer = model->addOperand(&type5); in CreateModel_dynamic_output_shape() local
107 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
111 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm2.model.cpp | 38 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
49 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
53 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
96 auto scratch_buffer = model->addOperand(&type5); in CreateModel_dynamic_output_shape() local
107 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
111 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm3.model.cpp | 39 auto scratch_buffer = model->addOperand(&type10); in CreateModel() local
50 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
54 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
99 auto scratch_buffer = model->addOperand(&type11); in CreateModel_dynamic_output_shape() local
110 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
114 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm3_state2_float16.model.cpp | 39 auto scratch_buffer = model->addOperand(&type10); in CreateModel() local
50 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
54 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
99 auto scratch_buffer = model->addOperand(&type11); in CreateModel_dynamic_output_shape() local
110 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
114 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm3_float16.model.cpp | 39 auto scratch_buffer = model->addOperand(&type10); in CreateModel() local
50 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
54 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
99 auto scratch_buffer = model->addOperand(&type11); in CreateModel_dynamic_output_shape() local
110 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
114 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm3_state_float16.model.cpp | 39 auto scratch_buffer = model->addOperand(&type10); in CreateModel() local
50 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
54 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
99 auto scratch_buffer = model->addOperand(&type11); in CreateModel_dynamic_output_shape() local
110 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
114 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm2_state2_relaxed.model.cpp | 38 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
49 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
53 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
98 auto scratch_buffer = model->addOperand(&type5); in CreateModel_dynamic_output_shape() local
109 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
113 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm2_state_float16.model.cpp | 38 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
49 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
53 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
96 auto scratch_buffer = model->addOperand(&type5); in CreateModel_dynamic_output_shape() local
107 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
111 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm3_state3.model.cpp | 39 auto scratch_buffer = model->addOperand(&type10); in CreateModel() local
50 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
54 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
99 auto scratch_buffer = model->addOperand(&type11); in CreateModel_dynamic_output_shape() local
110 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
114 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm3_state3_float16.model.cpp | 39 auto scratch_buffer = model->addOperand(&type10); in CreateModel() local
50 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
54 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
99 auto scratch_buffer = model->addOperand(&type11); in CreateModel_dynamic_output_shape() local
110 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
114 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm3_state_relaxed.model.cpp | 39 auto scratch_buffer = model->addOperand(&type10); in CreateModel() local
50 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
54 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
101 auto scratch_buffer = model->addOperand(&type11); in CreateModel_dynamic_output_shape() local
112 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
116 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm2_state_relaxed.model.cpp | 38 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
49 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
53 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
98 auto scratch_buffer = model->addOperand(&type5); in CreateModel_dynamic_output_shape() local
109 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
113 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm_float16.model.cpp | 38 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
49 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
53 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
96 auto scratch_buffer = model->addOperand(&type5); in CreateModel_dynamic_output_shape() local
107 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
111 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm_state2_relaxed.model.cpp | 38 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
49 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
53 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
98 auto scratch_buffer = model->addOperand(&type5); in CreateModel_dynamic_output_shape() local
109 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
113 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm3_state.model.cpp | 39 auto scratch_buffer = model->addOperand(&type10); in CreateModel() local
50 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
54 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
99 auto scratch_buffer = model->addOperand(&type11); in CreateModel_dynamic_output_shape() local
110 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
114 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm2_relaxed.model.cpp | 38 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
49 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
53 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
98 auto scratch_buffer = model->addOperand(&type5); in CreateModel_dynamic_output_shape() local
109 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
113 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm3_state2_relaxed.model.cpp | 39 auto scratch_buffer = model->addOperand(&type10); in CreateModel() local
50 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
54 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
101 auto scratch_buffer = model->addOperand(&type11); in CreateModel_dynamic_output_shape() local
112 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
116 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm_state_relaxed.model.cpp | 38 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
49 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
53 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
98 auto scratch_buffer = model->addOperand(&type5); in CreateModel_dynamic_output_shape() local
109 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
113 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm_relaxed.model.cpp | 38 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
49 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
53 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
98 auto scratch_buffer = model->addOperand(&type5); in CreateModel_dynamic_output_shape() local
109 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
113 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|
| D | lstm_state2_float16.model.cpp | 38 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
49 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
53 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
96 auto scratch_buffer = model->addOperand(&type5); in CreateModel_dynamic_output_shape() local
107 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel_dynamic_output_shape()
111 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel_dynamic_output_shape()
|