Searched defs:recurrentWeightsShape (Results 1 – 2 of 2) sorted by relevance
122 const T* recurrentWeightsData, const Shape& recurrentWeightsShape, in RNNStep()144 const Shape& recurrentWeightsShape, const int32_t activation, in RNNStep()
72 Shape recurrentWeightsShape = context->getInputShape(kRecurrentWeightsTensor); in executeTyped() local