Searched refs:kPrevCellStateTensor (Results 1 – 4 of 4) 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()
73 constexpr uint32_t kPrevCellStateTensor = 19; variable172 kPrevCellStateTensor, in prepare()302 const Shape cellStateShape = context->getInputShape(kPrevCellStateTensor); in prepare()357 const Shape prevCellStateShape = context->getInputShape(kPrevCellStateTensor); in prepare()390 const Shape prevCellStateShape = context->getInputShape(kPrevCellStateTensor); in execute()460 reinterpret_cast<const int16_t*>(context->getInputBuffer(kPrevCellStateTensor)); in execute()
57 static constexpr int kPrevCellStateTensor = 13; variable
238 prevCellState_ = GetInput(operation, operands, kPrevCellStateTensor); in QuantizedLSTMCell()314 auto prevCellState = GetInput(operation, operands, kPrevCellStateTensor); in prepare()