Home
last modified time | relevance | path

Searched refs:scratch_buffer (Results 1 – 22 of 22) sorted by relevance

/frameworks/ml/nn/runtime/test/specs/
Dlstm2_state.mod.py58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell * … variable
94 ).To([scratch_buffer, output_state_out, cell_state_out, output])
136 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
Dlstm2.mod.py58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell * … variable
94 ).To([scratch_buffer, output_state_out, cell_state_out, output])
136 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
Dlstm.mod.py58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, (n_cell *… variable
94 ).To([scratch_buffer, output_state_out, cell_state_out, output])
144 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
Dlstm3_state2.mod.py58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, (n_cell *… variable
95 ).To([scratch_buffer, output_state_out, cell_state_out, output])
648 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
Dlstm3_state.mod.py58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, (n_cell *… variable
95 ).To([scratch_buffer, output_state_out, cell_state_out, output])
648 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
Dlstm2_state2.mod.py58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, n_cell * … variable
94 ).To([scratch_buffer, output_state_out, cell_state_out, output])
136 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
Dlstm_state.mod.py58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, (n_cell *… variable
94 ).To([scratch_buffer, output_state_out, cell_state_out, output])
144 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
Dlstm_state2.mod.py58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, (n_cell *… variable
94 ).To([scratch_buffer, output_state_out, cell_state_out, output])
144 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
Dlstm3_state3.mod.py58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, (n_cell *… variable
95 ).To([scratch_buffer, output_state_out, cell_state_out, output])
648 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
Dlstm3.mod.py58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, (n_cell *… variable
95 ).To([scratch_buffer, output_state_out, cell_state_out, output])
628 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
/frameworks/ml/nn/tools/test_generator/tests/P_lstm/
Dlstm.mod.py58 scratch_buffer = IgnoredOutput("scratch_buffer", "TENSOR_FLOAT32", "{%d, %d}" % (n_batch, (n_cell *… variable
94 ).To([scratch_buffer, output_state_out, cell_state_out, output])
153 scratch_buffer: [ 0 for x in range(n_batch * n_cell * 4) ],
Dstdout.txt.expect37 auto scratch_buffer = model->addOperand(&type9);
42 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_…
46 {scratch_buffer, output_state_out, cell_state_out, output});
/frameworks/ml/nn/runtime/test/generated/models/
Dlstm2.model.cpp37 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
42 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
46 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
Dlstm3_state2.model.cpp38 auto scratch_buffer = model->addOperand(&type10); in CreateModel() local
43 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
47 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
Dlstm_state2.model.cpp37 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
42 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
46 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
Dlstm3_state.model.cpp38 auto scratch_buffer = model->addOperand(&type10); in CreateModel() local
43 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
47 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
Dlstm.model.cpp37 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
42 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
46 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
Dlstm3_state3.model.cpp38 auto scratch_buffer = model->addOperand(&type10); in CreateModel() local
43 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
47 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
Dlstm3.model.cpp38 auto scratch_buffer = model->addOperand(&type10); in CreateModel() local
43 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
47 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
Dlstm2_state.model.cpp37 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
42 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
46 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
Dlstm_state.model.cpp37 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
42 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
46 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()
Dlstm2_state2.model.cpp37 auto scratch_buffer = model->addOperand(&type9); in CreateModel() local
42 …cell_state_in, activation_param, cell_clip_param, proj_clip_param}, {scratch_buffer, output_state_… in CreateModel()
46 {scratch_buffer, output_state_out, cell_state_out, output}); in CreateModel()