Searched refs:kOutputStateInTensor (Results 1 – 4 of 4) sorted by relevance
78 kOutputStateInTensor, in prepare()211 const Shape outputStateShape = context->getInputShape(kOutputStateInTensor); in prepare()275 Shape outputStateOutTensor = context->getInputShape(kOutputStateInTensor); in prepare()292 const auto outputStateSize = getNumberOfElements(context->getInputShape(kOutputStateInTensor)); in execute()343 context->getInputBuffer<float>(kOutputStateInTensor), in execute()396 context->getInputBuffer<_Float16>(kOutputStateInTensor), in execute()
83 output_state_in_ = GetInput(operation, operands, kOutputStateInTensor); in LSTMCell()320 kOutputStateInTensor, in Prepare()
59 constexpr uint32_t kOutputStateInTensor = 18; variable
87 static constexpr int kOutputStateInTensor = 18; variable