Searched refs:kHiddenStateTensor (Results 1 – 1 of 1) sorted by relevance
36 constexpr uint32_t kHiddenStateTensor = 4; variable73 const T* hiddenState = context->getInputBuffer<T>(kHiddenStateTensor); in executeTyped()158 Shape hiddenState = context->getInputShape(kHiddenStateTensor); in prepare()190 Shape outputStateShape = context->getInputShape(kHiddenStateTensor); in prepare()