Searched refs:kCellStateInTensor (Results 1 – 3 of 3) sorted by relevance
66 constexpr uint32_t kCellStateInTensor = 19; variable173 kCellStateInTensor, in prepare()308 const Shape cellStateShape = context->getInputShape(kCellStateInTensor); in prepare()369 const auto cellStateSize = getNumberOfElements(context->getInputShape(kCellStateInTensor)); in execute()407 context->getInputBuffer<float>(kCellStateInTensor), in execute()449 context->getInputBuffer<_Float16>(kCellStateInTensor), in execute()
87 static constexpr int kCellStateInTensor = 19; variable
79 cell_state_in_ = GetInput(operation, operands, kCellStateInTensor); in LSTMCell()