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