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()
|