Searched refs:kBwRecurrentWeightsTensor (Results 1 – 1 of 1) sorted by relevance
39 constexpr uint32_t kBwRecurrentWeightsTensor = 6; variable134 const T* bwRecurrentWeights = context->getInputBuffer<T>(kBwRecurrentWeightsTensor); in executeTyped()135 Shape bwRecurrentWeightsShape = context->getInputShape(kBwRecurrentWeightsTensor); in executeTyped()359 kFwHiddenStateTensor, kBwWeightsTensor, kBwRecurrentWeightsTensor, kBwBiasTensor, in prepare()373 Shape bwRecurrentWeights = context->getInputShape(kBwRecurrentWeightsTensor); in prepare()