| /frameworks/ml/nn/runtime/test/generated/models/ |
| svdf.model.cpp | 18 auto state_out = model->addOperand(&type4); local
21 model->addOperation(ANEURALNETWORKS_SVDF, {input, weights_feature, weights_time, bias, state_in, rank_param, activation_param}, {state_out, output});
25 {state_out, output});
|
| svdf_state.model.cpp | 18 auto state_out = model->addOperand(&type4); local
21 model->addOperation(ANEURALNETWORKS_SVDF, {input, weights_feature, weights_time, bias, state_in, rank_param, activation_param}, {state_out, output});
25 {state_out, output});
|
| /frameworks/ml/nn/runtime/test/specs/ |
| 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 : [
|
| 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)],
|
