Searched refs:kRecurrentWeightsTensor (Results 1 – 3 of 3) sorted by relevance
34 constexpr uint32_t kRecurrentWeightsTensor = 2; variable70 const T* recurrentWeights = context->getInputBuffer<T>(kRecurrentWeightsTensor); in executeTyped()71 Shape recurrentWeightsShape = context->getInputShape(kRecurrentWeightsTensor); in executeTyped()156 Shape recurrentWeights = context->getInputShape(kRecurrentWeightsTensor); in prepare()
41 static constexpr int kRecurrentWeightsTensor = 2; variable
34 recurrent_weights_ = GetInput(operation, operands, kRecurrentWeightsTensor); in RNN()56 GetInput(operation, operands, kRecurrentWeightsTensor); in Prepare()