Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceRNN.cpp40 constexpr uint32_t kFwAuxWeightsTensor = 10; // optional variable
102 fwAuxWeights = context->getInputBuffer<T>(kFwAuxWeightsTensor); in executeTyped()
106 Shape fwAuxWeightsShape = context->getInputShape(kFwAuxWeightsTensor); in executeTyped()
302 Shape fwAuxWeights = context->getInputShape(kFwAuxWeightsTensor); in prepare()
306 context->isOmittedInput(kFwAuxWeightsTensor) && in prepare()
309 !context->isOmittedInput(kFwAuxWeightsTensor) && in prepare()