Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceRNN.cpp41 constexpr uint32_t kBwAuxWeightsTensor = 11; // optional variable
103 bwAuxWeights = context->getInputBuffer<T>(kBwAuxWeightsTensor); in executeTyped()
107 Shape bwAuxWeightsShape = context->getInputShape(kBwAuxWeightsTensor); in executeTyped()
303 Shape bwAuxWeights = context->getInputShape(kBwAuxWeightsTensor); in prepare()
307 context->isOmittedInput(kBwAuxWeightsTensor)) || in prepare()
310 !context->isOmittedInput(kBwAuxWeightsTensor)); in prepare()