Searched refs:rank_param (Results 1 - 12 of 12) sorted by relevance

/frameworks/ml/nn/runtime/test/specs/V1_0/
H A Dsvdf_state.mod.py29 rank_param = Int32Scalar("rank_param", 1) variable
35 rank_param, activation_param).To([state_out, output])
H A Dsvdf.mod.py31 rank_param = Int32Scalar("rank_param", rank) variable
37 rank_param, activation_param).To([state_out, output])
H A Dsvdf2.mod.py31 rank_param = Int32Scalar("rank_param", rank) variable
37 rank_param, activation_param).To([state_out, output])
/frameworks/ml/nn/runtime/test/specs/V1_1/
H A Dsvdf_state_relaxed.mod.py29 rank_param = Int32Scalar("rank_param", 1) variable
35 rank_param, activation_param).To([state_out, output])
H A Dsvdf2_relaxed.mod.py31 rank_param = Int32Scalar("rank_param", rank) variable
37 rank_param, activation_param).To([state_out, output])
H A Dsvdf_relaxed.mod.py31 rank_param = Int32Scalar("rank_param", rank) variable
37 rank_param, activation_param).To([state_out, output])
/frameworks/ml/nn/runtime/test/generated/models/
H A Dsvdf.model.cpp16 auto rank_param = model->addOperand(&type5); local
22 model->setOperandValue(rank_param, rank_param_init, sizeof(int32_t) * 1);
25 model->addOperation(ANEURALNETWORKS_SVDF, {input, weights_feature, weights_time, bias, state_in, rank_param, activation_param}, {state_out, output});
H A Dsvdf2.model.cpp16 auto rank_param = model->addOperand(&type5); local
22 model->setOperandValue(rank_param, rank_param_init, sizeof(int32_t) * 1);
25 model->addOperation(ANEURALNETWORKS_SVDF, {input, weights_feature, weights_time, bias, state_in, rank_param, activation_param}, {state_out, output});
H A Dsvdf2_relaxed.model.cpp16 auto rank_param = model->addOperand(&type5); local
22 model->setOperandValue(rank_param, rank_param_init, sizeof(int32_t) * 1);
25 model->addOperation(ANEURALNETWORKS_SVDF, {input, weights_feature, weights_time, bias, state_in, rank_param, activation_param}, {state_out, output});
H A Dsvdf_relaxed.model.cpp16 auto rank_param = model->addOperand(&type5); local
22 model->setOperandValue(rank_param, rank_param_init, sizeof(int32_t) * 1);
25 model->addOperation(ANEURALNETWORKS_SVDF, {input, weights_feature, weights_time, bias, state_in, rank_param, activation_param}, {state_out, output});
H A Dsvdf_state.model.cpp16 auto rank_param = model->addOperand(&type5); local
22 model->setOperandValue(rank_param, rank_param_init, sizeof(int32_t) * 1);
25 model->addOperation(ANEURALNETWORKS_SVDF, {input, weights_feature, weights_time, bias, state_in, rank_param, activation_param}, {state_out, output});
H A Dsvdf_state_relaxed.model.cpp16 auto rank_param = model->addOperand(&type5); local
22 model->setOperandValue(rank_param, rank_param_init, sizeof(int32_t) * 1);
25 model->addOperation(ANEURALNETWORKS_SVDF, {input, weights_feature, weights_time, bias, state_in, rank_param, activation_param}, {state_out, output});

Completed in 93 milliseconds