Searched refs:kOutputStateInTensor (Results 1 – 3 of 3) sorted by relevance
66 constexpr uint32_t kOutputStateInTensor = 18; variable185 kOutputStateInTensor, in prepare()317 const Shape outputStateShape = context->getInputShape(kOutputStateInTensor); in prepare()381 Shape outputStateOutTensor = context->getInputShape(kOutputStateInTensor); in prepare()398 const auto outputStateSize = getNumberOfElements(context->getInputShape(kOutputStateInTensor)); in execute()449 context->getInputBuffer<float>(kOutputStateInTensor), in execute()502 context->getInputBuffer<_Float16>(kOutputStateInTensor), in execute()
88 static constexpr int kOutputStateInTensor = 18; variable
83 output_state_in_ = GetInput(operation, operands, kOutputStateInTensor); in LSTMCell()320 kOutputStateInTensor, in Prepare()