Searched refs:memory_size_ (Results 1 – 1 of 1) sorted by relevance
171 memory_size_(memory_size), in SVDFOpModel()176 {units_ * rank_, memory_size_}, // weights_time tensor in SVDFOpModel()199 std::vector<std::vector<uint32_t>> output_shapes{{batches_, memory_size_ * units_ * rank_}, in SVDFOpModel()214 StateIn_.insert(StateIn_.end(), batches_ * units_ * rank_ * memory_size_, 0.f); in SVDFOpModel()303 const uint32_t memory_size_; member in android::nn::wrapper::SVDFOpModel