Searched refs:kBwRecurrentWeightsTensor (Results 1 – 2 of 2) sorted by relevance
105 const T* bwRecurrentWeights = context->getInputBuffer<T>(kBwRecurrentWeightsTensor); in executeTyped()106 Shape bwRecurrentWeightsShape = context->getInputShape(kBwRecurrentWeightsTensor); in executeTyped()300 kFwHiddenStateTensor, kBwWeightsTensor, kBwRecurrentWeightsTensor, kBwBiasTensor, in prepare()314 Shape bwRecurrentWeights = context->getInputShape(kBwRecurrentWeightsTensor); in prepare()
33 constexpr uint32_t kBwRecurrentWeightsTensor = 6; variable