Searched refs:prevOutput_ (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | QuantizedLSTM.cpp | 242 prevOutput_ = GetInput(operation, operands, kPrevOutputTensor); in QuantizedLSTMCell()
381 weightsShape.dimensions = {4 * SizeOfDimension(prevOutput_, 1), in eval()
382 SizeOfDimension(input_, 1) + SizeOfDimension(prevOutput_, 1)}; in eval()
389 concatenateBiases(SizeOfDimension(prevOutput_, 1), bias.data()); in eval()
416 GetBuffer<const uint8_t>(prevOutput_), convertShapeToDims(prevOutput_->shape()), in eval()
|
| D | QuantizedLSTMTest.cpp | 78 &prevOutput_); in QuantizedLSTMOpModel()
136 ASSERT_EQ(setInputTensor(&execution, QuantizedLSTMCell::kPrevOutputTensor, prevOutput_), in invoke()
148 prevOutput_ = output_; in invoke()
216 std::vector<uint8_t> prevOutput_; member in android::nn::wrapper::QuantizedLSTMOpModel
|
| D | QuantizedLSTM.h | 83 const RunTimeOperandInfo* prevOutput_; variable
|