Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceRNN.cpp30 constexpr uint32_t kFwRecurrentWeightsTensor = 2; variable
84 const T* fwRecurrentWeights = context->getInputBuffer<T>(kFwRecurrentWeightsTensor); in executeTyped()
85 Shape fwRecurrentWeightsShape = context->getInputShape(kFwRecurrentWeightsTensor); in executeTyped()
282 kInputTensor, kFwWeightsTensor, kFwRecurrentWeightsTensor, kFwBiasTensor, in prepare()
293 Shape fwRecurrentWeights = context->getInputShape(kFwRecurrentWeightsTensor); in prepare()