Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceRNN.cpp96 const T* fwWeights = context->getInputBuffer<T>(kFwWeightsTensor); in executeTyped()
97 Shape fwWeightsShape = context->getInputShape(kFwWeightsTensor); in executeTyped()
299 kInputTensor, kFwWeightsTensor, kFwRecurrentWeightsTensor, kFwBiasTensor, in prepare()
309 Shape fwWeights = context->getInputShape(kFwWeightsTensor); in prepare()
/packages/modules/NeuralNetworks/common/types/operations/include/
DBidirectionalSequenceRNN.h27 constexpr uint32_t kFwWeightsTensor = 1; variable