Searched refs:kRankParam (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | SVDF.cpp | 41 const auto& rankOperand = *GetInput(operation, operands, kRankParam); in SVDF()
60 kRankParam, kActivationParam, in Prepare()
69 const auto& rankOperand = *GetInput(operation, operands, kRankParam); in Prepare()
82 const int rank = getScalarData<int>(*GetInput(operation, operands, kRankParam)); in Prepare()
|
| D | SVDF.h | 52 static constexpr int kRankParam = 5; variable
|
| D | SVDFTest.cpp | 262 ASSERT_EQ(execution.setInput(SVDF::kRankParam, &rank_, sizeof(rank_)), Result::NO_ERROR); in Invoke()
|