Searched refs:kBwWeightsTensor (Results 1 – 1 of 1) sorted by relevance
38 constexpr uint32_t kBwWeightsTensor = 5; variable132 const T* bwWeights = context->getInputBuffer<T>(kBwWeightsTensor); in executeTyped()133 Shape bwWeightsShape = context->getInputShape(kBwWeightsTensor); in executeTyped()359 kFwHiddenStateTensor, kBwWeightsTensor, kBwRecurrentWeightsTensor, kBwBiasTensor, in prepare()372 Shape bwWeights = context->getInputShape(kBwWeightsTensor); in prepare()