Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceRNN.cpp29 constexpr uint32_t kFwWeightsTensor = 1; variable
82 const T* fwWeights = context->getInputBuffer<T>(kFwWeightsTensor); in executeTyped()
83 Shape fwWeightsShape = context->getInputShape(kFwWeightsTensor); in executeTyped()
282 kInputTensor, kFwWeightsTensor, kFwRecurrentWeightsTensor, kFwBiasTensor, in prepare()
292 Shape fwWeights = context->getInputShape(kFwWeightsTensor); in prepare()