Searched refs:input_size_ (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | RNNTest.cpp | 134 : batches_(batches), units_(units), input_size_(size), activation_(kActivationRelu) { in BasicRNNOpModel()
137 OperandType InputTy(Type::TENSOR_FLOAT32, {batches_, input_size_}); in BasicRNNOpModel()
139 OperandType WeightTy(Type::TENSOR_FLOAT32, {units_, input_size_}); in BasicRNNOpModel()
156 Input_.insert(Input_.end(), batches_ * input_size_, 0.f); in BasicRNNOpModel()
187 uint32_t input_size() const { return input_size_; } in input_size()
226 const uint32_t input_size_; member in android::nn::wrapper::BasicRNNOpModel
|
| D | SVDFTest.cpp | 170 input_size_(input_size), in SVDFOpModel()
174 {batches_, input_size_}, // Input tensor in SVDFOpModel()
175 {units_ * rank_, input_size_}, // weights_feature tensor in SVDFOpModel()
213 Input_.insert(Input_.end(), batches_ * input_size_, 0.f); in SVDFOpModel()
293 int input_size() const { return input_size_; } in input_size()
302 const uint32_t input_size_; member in android::nn::wrapper::SVDFOpModel
|