Searched defs:auxInputShape (Results 1 – 2 of 2) sorted by relevance
137 const Shape& auxInputShape, const T* hiddenStateInputData, const T* biasData, in RNNStep()
154 Shape auxInputShape = context->getInputShape(kAuxInputTensor); in executeTyped() local