Searched refs:kPrevCellStateTensor (Results 1 – 4 of 4) sorted by relevance
57 static constexpr int kPrevCellStateTensor = 13; variable
62 inputOperandTypeParams[QuantizedLSTMCell::kPrevCellStateTensor].shape[1]; in QuantizedLSTMOpModel()79 initializeInputData(inputOperandTypeParams[QuantizedLSTMCell::kPrevCellStateTensor], in QuantizedLSTMOpModel()134 setInputTensor(&execution, QuantizedLSTMCell::kPrevCellStateTensor, prevCellState_), in invoke()
70 constexpr uint32_t kPrevCellStateTensor = 19; variable171 kPrevCellStateTensor, in prepare()301 const Shape cellStateShape = context->getInputShape(kPrevCellStateTensor); in prepare()356 const Shape prevCellStateShape = context->getInputShape(kPrevCellStateTensor); in prepare()388 const Shape prevCellStateShape = context->getInputShape(kPrevCellStateTensor); in execute()458 reinterpret_cast<const int16_t*>(context->getInputBuffer(kPrevCellStateTensor)); in execute()
241 prevCellState_ = GetInput(operation, operands, kPrevCellStateTensor); in QuantizedLSTMCell()317 auto prevCellState = GetInput(operation, operands, kPrevCellStateTensor); in prepare()