Searched refs:prevOutput_ (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | QuantizedLSTM.cpp | 243 prevOutput_ = GetInput(operation, operands, kPrevOutputTensor); in QuantizedLSTMCell()
382 weightsShape.dimensions = {4 * SizeOfDimension(prevOutput_, 1), in eval()
383 SizeOfDimension(input_, 1) + SizeOfDimension(prevOutput_, 1)}; in eval()
390 concatenateBiases(SizeOfDimension(prevOutput_, 1), bias.data()); in eval()
417 GetBuffer<const uint8_t>(prevOutput_), convertShapeToDims(prevOutput_->shape()), in eval()
|
| D | QuantizedLSTMTest.cpp | 79 &prevOutput_); in QuantizedLSTMOpModel()
137 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kPrevOutputTensor, prevOutput_), in invoke()
149 prevOutput_ = output_; in invoke()
217 std::vector<uint8_t> prevOutput_; member in android::nn::wrapper::QuantizedLSTMOpModel
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | QuantizedLSTM.h | 83 const RunTimeOperandInfo* prevOutput_; variable
|