Searched refs:kRecurrentWeightsTensor (Results 1 – 3 of 3) sorted by relevance
29 constexpr uint32_t kRecurrentWeightsTensor = 2; variable62 const T* recurrentWeights = context->getInputBuffer<T>(kRecurrentWeightsTensor); in executeTyped()63 Shape recurrentWeightsShape = context->getInputShape(kRecurrentWeightsTensor); in executeTyped()134 Shape recurrentWeights = context->getInputShape(kRecurrentWeightsTensor); in prepare()
39 static constexpr int kRecurrentWeightsTensor = 2; variable
33 recurrent_weights_ = GetInput(operation, operands, kRecurrentWeightsTensor); in RNN()59 GetInput(operation, operands, kRecurrentWeightsTensor); in Prepare()