Searched refs:kFwHiddenStateTensor (Results 1 – 2 of 2) sorted by relevance
101 const T* fwHiddenState = context->getInputBuffer<T>(kFwHiddenStateTensor); in executeTyped()300 kFwHiddenStateTensor, kBwWeightsTensor, kBwRecurrentWeightsTensor, kBwBiasTensor, in prepare()312 Shape fwHiddenState = context->getInputShape(kFwHiddenStateTensor); in prepare()399 context->getInputShape(kFwHiddenStateTensor))); in prepare()
30 constexpr uint32_t kFwHiddenStateTensor = 4; variable