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