Searched refs:hidden_state_out (Results 1 - 4 of 4) sorted by relevance
| /frameworks/ml/nn/runtime/test/specs/ |
| H A D | rnn_state.mod.py | 31 hidden_state_out = IgnoredOutput("hidden_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (batches, units)) variable
35 activation_param).To([hidden_state_out, output])
106 hidden_state_out : [
|
| H A D | rnn.mod.py | 31 hidden_state_out = IgnoredOutput("hidden_state_out", "TENSOR_FLOAT32", "{%d, %d}" % (batches, units)) variable
35 activation_param).To([hidden_state_out, output])
196 hidden_state_out: [0 for x in range(batches * units)],
|
| /frameworks/ml/nn/runtime/test/generated/models/ |
| H A D | rnn.model.cpp | 16 auto hidden_state_out = model->addOperand(&type4); local
19 model->addOperation(ANEURALNETWORKS_RNN, {input, weights, recurrent_weights, bias, hidden_state_in, activation_param}, {hidden_state_out, output});
23 {hidden_state_out, output});
|
| H A D | rnn_state.model.cpp | 16 auto hidden_state_out = model->addOperand(&type4); local
19 model->addOperation(ANEURALNETWORKS_RNN, {input, weights, recurrent_weights, bias, hidden_state_in, activation_param}, {hidden_state_out, output});
23 {hidden_state_out, output});
|
Completed in 8 milliseconds