Lines Matching refs:model_
139 inputs.push_back(model_.addOperand(&InputTy)); in BasicRNNOpModel()
141 inputs.push_back(model_.addOperand(&WeightTy)); in BasicRNNOpModel()
143 inputs.push_back(model_.addOperand(&RecurrentWeightTy)); in BasicRNNOpModel()
145 inputs.push_back(model_.addOperand(&BiasTy)); in BasicRNNOpModel()
147 inputs.push_back(model_.addOperand(&HiddenStateTy)); in BasicRNNOpModel()
149 inputs.push_back(model_.addOperand(&ActionParamTy)); in BasicRNNOpModel()
153 outputs.push_back(model_.addOperand(&HiddenStateTy)); in BasicRNNOpModel()
155 outputs.push_back(model_.addOperand(&OutputTy)); in BasicRNNOpModel()
162 model_.addOperation(ANEURALNETWORKS_RNN, inputs, outputs); in BasicRNNOpModel()
163 model_.identifyInputsAndOutputs(inputs, outputs); in BasicRNNOpModel()
165 model_.finish(); in BasicRNNOpModel()
193 ASSERT_TRUE(model_.isValid()); in Invoke()
197 Compilation compilation(&model_); in Invoke()
223 Model model_; member in android::nn::wrapper::BasicRNNOpModel