Searched refs:kPrevCellStateTensor (Results 1 – 5 of 5) sorted by relevance
63 inputOperandTypeParams[QuantizedLSTMCell::kPrevCellStateTensor].shape[1]; in QuantizedLSTMOpModel()80 initializeInputData(inputOperandTypeParams[QuantizedLSTMCell::kPrevCellStateTensor], in QuantizedLSTMOpModel()135 setInputTensor(&execution, QuantizedLSTMCell::kPrevCellStateTensor, prevCellState_), in invoke()
56 kPrevCellStateTensor, in prepare()186 const Shape cellStateShape = context->getInputShape(kPrevCellStateTensor); in prepare()241 const Shape prevCellStateShape = context->getInputShape(kPrevCellStateTensor); in prepare()274 const Shape prevCellStateShape = context->getInputShape(kPrevCellStateTensor); in execute()344 reinterpret_cast<const int16_t*>(context->getInputBuffer(kPrevCellStateTensor)); in execute()
242 prevCellState_ = GetInput(operation, operands, kPrevCellStateTensor); in QuantizedLSTMCell()318 auto prevCellState = GetInput(operation, operands, kPrevCellStateTensor); in prepare()
57 static constexpr int kPrevCellStateTensor = 13; variable
63 constexpr uint32_t kPrevCellStateTensor = 19; variable