Searched refs:rank_ (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | SVDFTest.cpp | 174 rank_(rank) { in SVDFOpModel()
177 {units_ * rank_, input_size_}, // weights_feature tensor in SVDFOpModel()
178 {units_ * rank_, memory_size_}, // weights_time tensor in SVDFOpModel()
180 {batches_, memory_size * units_ * rank_}, // state in tensor in SVDFOpModel()
201 std::vector<std::vector<uint32_t>> output_shapes{{batches_, memory_size_ * units_ * rank_}, in SVDFOpModel()
216 StateIn_.insert(StateIn_.end(), batches_ * units_ * rank_ * memory_size_, 0.f); in SVDFOpModel()
264 ASSERT_EQ(execution.setInput(SVDF::kRankParam, &rank_, sizeof(rank_)), Result::NO_ERROR); in Invoke()
306 const uint32_t rank_; member in android::nn::wrapper::SVDFOpModel
|
| D | SVDF.cpp | 42 params_.rank_ = getScalarDataWithDefault<int>(rankOperand, 0); in SVDF()
169 const int rank = params_.rank_; in EvalFloat32()
|
| /packages/modules/Bluetooth/system/bta/csis/ |
| D | csis_types.h | 165 rank_(kUnknownRank), in CsisInstance()
177 uint8_t GetRank(void) const { return rank_; } in GetRank()
179 log::debug("current rank: {}, new rank: {}", static_cast<int>(rank_), in SetRank()
181 rank_ = rank; in SetRank()
203 uint8_t rank_; variable
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | SVDF.h | 32 int rank_; member
|