Searched refs:input_size_ (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | RNNTest.cpp | 135 : batches_(batches), units_(units), input_size_(size), activation_(kActivationRelu) { in BasicRNNOpModel()
138 OperandType InputTy(Type::TENSOR_FLOAT32, {batches_, input_size_}); in BasicRNNOpModel()
140 OperandType WeightTy(Type::TENSOR_FLOAT32, {units_, input_size_}); in BasicRNNOpModel()
157 Input_.insert(Input_.end(), batches_ * input_size_, 0.f); in BasicRNNOpModel()
188 uint32_t input_size() const { return input_size_; } in input_size()
227 const uint32_t input_size_; member in android::nn::wrapper::BasicRNNOpModel
|
| D | SVDFTest.cpp | 172 input_size_(input_size), in SVDFOpModel()
176 {batches_, input_size_}, // Input tensor in SVDFOpModel()
177 {units_ * rank_, input_size_}, // weights_feature tensor in SVDFOpModel()
215 Input_.insert(Input_.end(), batches_ * input_size_, 0.f); in SVDFOpModel()
295 int input_size() const { return input_size_; } in input_size()
304 const uint32_t input_size_; member in android::nn::wrapper::SVDFOpModel
|