Searched refs:state_out (Results 1 – 5 of 5) sorted by relevance
| /frameworks/ml/nn/runtime/test/specs/ |
| D | svdf_state.mod.py | 31 state_out = Output("state_out", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*units)) variable
35 rank_param, activation_param).To([state_out, output])
88 state_out : [
|
| D | svdf.mod.py | 31 state_out = IgnoredOutput("state_out", "TENSOR_FLOAT32", "{%d, %d}" % (batches, memory_size*units)) variable
35 rank_param, activation_param).To([state_out, output])
135 output0 = {state_out:[0 for x in range(batches * (memory_size - 1) * units)],
|
| /frameworks/ml/nn/runtime/test/generated/models/ |
| D | svdf_state.model.cpp | 18 auto state_out = model->addOperand(&type4); in CreateModel() local
21 …weights_feature, weights_time, bias, state_in, rank_param, activation_param}, {state_out, output}); in CreateModel()
25 {state_out, output}); in CreateModel()
|
| D | svdf.model.cpp | 18 auto state_out = model->addOperand(&type4); in CreateModel() local
21 …weights_feature, weights_time, bias, state_in, rank_param, activation_param}, {state_out, output}); in CreateModel()
25 {state_out, output}); in CreateModel()
|
| /frameworks/ml/nn/common/operations/ |
| D | SVDF.cpp | 30 float* state_out) { in svdf_right_shift_state() argument
32 state_out[i] = state_in[i + 1]; in svdf_right_shift_state()
34 state_out[state_len - 1] = shift_value; in svdf_right_shift_state()
|