Searched refs:kRankParam (Results 1 – 3 of 3) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | SVDF.cpp | 39 const auto& rankOperand = *GetInput(operation, operands, kRankParam); in SVDF()
58 kRankParam, kActivationParam, in Prepare()
67 const auto& rankOperand = *GetInput(operation, operands, kRankParam); in Prepare()
80 const int rank = getScalarData<int>(*GetInput(operation, operands, kRankParam)); in Prepare()
|
| H A D | SVDF.h | 52 static constexpr int kRankParam = 5; variable
|
| H A D | SVDFTest.cpp | 264 ASSERT_EQ(execution.setInput(SVDF::kRankParam, &rank_, sizeof(rank_)), Result::NO_ERROR); in Invoke()
|