Home
last modified time | relevance | path

Searched refs:kFwHiddenStateTensor (Results 1 – 1 of 1) sorted by relevance

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceRNN.cpp32 constexpr uint32_t kFwHiddenStateTensor = 4; variable
87 const T* fwHiddenState = context->getInputBuffer<T>(kFwHiddenStateTensor); in executeTyped()
283 kFwHiddenStateTensor, kBwWeightsTensor, kBwRecurrentWeightsTensor, kBwBiasTensor, in prepare()
295 Shape fwHiddenState = context->getInputShape(kFwHiddenStateTensor); in prepare()