Searched refs:cellStateOut_ (Results 1 – 3 of 3) sorted by relevance
82 cellStateOut_.resize(numBatches * outputSize, 0); in QuantizedLSTMOpModel()140 setOutputTensor(&execution, QuantizedLSTMCell::kCellStateOutTensor, &cellStateOut_), in invoke()149 prevCellState_ = cellStateOut_; in invoke()218 std::vector<int16_t> cellStateOut_; member in android::nn::wrapper::QuantizedLSTMOpModel
85 RunTimeOperandInfo* cellStateOut_; variable
244 cellStateOut_ = GetOutput(operation, operands, kCellStateOutTensor); in QuantizedLSTMCell()421 GetBuffer<int16_t>(cellStateOut_), convertShapeToDims(cellStateOut_->shape()), in eval()