Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceRNN.cpp35 constexpr uint32_t kBwRecurrentWeightsTensor = 6; variable
91 const T* bwRecurrentWeights = context->getInputBuffer<T>(kBwRecurrentWeightsTensor); in executeTyped()
92 Shape bwRecurrentWeightsShape = context->getInputShape(kBwRecurrentWeightsTensor); in executeTyped()
283 kFwHiddenStateTensor, kBwWeightsTensor, kBwRecurrentWeightsTensor, kBwBiasTensor, in prepare()
297 Shape bwRecurrentWeights = context->getInputShape(kBwRecurrentWeightsTensor); in prepare()