Home
last modified time | relevance | path

Searched refs:batches_ (Results 1 – 2 of 2) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DRNNTest.cpp135 : batches_(batches), units_(units), input_size_(size), activation_(kActivationRelu) { in BasicRNNOpModel()
138 OperandType InputTy(Type::TENSOR_FLOAT32, {batches_, input_size_}); in BasicRNNOpModel()
146 OperandType HiddenStateTy(Type::TENSOR_FLOAT32, {batches_, units_}); in BasicRNNOpModel()
154 OperandType OutputTy(Type::TENSOR_FLOAT32, {batches_, units_}); in BasicRNNOpModel()
157 Input_.insert(Input_.end(), batches_ * input_size_, 0.f); in BasicRNNOpModel()
158 HiddenStateIn_.insert(HiddenStateIn_.end(), batches_ * units_, 0.f); in BasicRNNOpModel()
159 HiddenStateOut_.insert(HiddenStateOut_.end(), batches_ * units_, 0.f); in BasicRNNOpModel()
160 Output_.insert(Output_.end(), batches_ * units_, 0.f); in BasicRNNOpModel()
190 uint32_t num_batches() const { return batches_; } in num_batches()
225 const uint32_t batches_; member in android::nn::wrapper::BasicRNNOpModel
DSVDFTest.cpp170 : batches_(batches), in SVDFOpModel()
176 {batches_, input_size_}, // Input 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()
202 {batches_, units_}}; in SVDFOpModel()
215 Input_.insert(Input_.end(), batches_ * input_size_, 0.f); in SVDFOpModel()
216 StateIn_.insert(StateIn_.end(), batches_ * units_ * rank_ * memory_size_, 0.f); in SVDFOpModel()
297 int num_batches() const { return batches_; } in num_batches()
302 const uint32_t batches_; member in android::nn::wrapper::SVDFOpModel