Searched refs:rank_ (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | SVDFTest.cpp | 192 rank_(rank) { in SVDFOpModel()
195 {units_ * rank_, input_size_}, // weights_feature tensor in SVDFOpModel()
196 {units_ * rank_, memory_size_}, // weights_time tensor in SVDFOpModel()
198 {batches_, memory_size * units_ * rank_}, // state in tensor in SVDFOpModel()
219 std::vector<std::vector<uint32_t>> output_shapes{{batches_, memory_size_ * units_ * rank_}, in SVDFOpModel()
234 StateIn_.insert(StateIn_.end(), batches_ * units_ * rank_ * memory_size_, 0.f); in SVDFOpModel()
282 ASSERT_EQ(execution.setInput(SVDF::kRankParam, &rank_, sizeof(rank_)), in Invoke()
328 const uint32_t rank_; member in android::nn::wrapper::SVDFOpModel
|
| D | SVDF.h | 30 int rank_; member
|
| D | SVDF.cpp | 37 params_.rank_ = getScalarData<int>(*GetInput(operation, operands, kRankParam)); in SVDF()
153 const int rank = params_.rank_; in EvalFloat32()
|