Searched refs:kBwWeightsTensor (Results 1 – 1 of 1) sorted by relevance
34 constexpr uint32_t kBwWeightsTensor = 5; variable89 const T* bwWeights = context->getInputBuffer<T>(kBwWeightsTensor); in executeTyped()90 Shape bwWeightsShape = context->getInputShape(kBwWeightsTensor); in executeTyped()283 kFwHiddenStateTensor, kBwWeightsTensor, kBwRecurrentWeightsTensor, kBwBiasTensor, in prepare()296 Shape bwWeights = context->getInputShape(kBwWeightsTensor); in prepare()