Searched refs:auxInputShape (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceRNN.cpp | 156 Shape auxInputShape = context->getInputShape(kAuxInputTensor); in executeTyped() local
184 auxInputTransposed.resize(getNumberOfElements(auxInputShape)); in executeTyped()
194 transposeFirstTwoDims(auxInput, auxInputShape, auxInputTransposed.data()); in executeTyped()
211 std::swap(auxInputShape.dimensions[0], auxInputShape.dimensions[1]); in executeTyped()
224 auxInputSize = getSizeOfDimension(auxInputShape, 2); in executeTyped()
230 Shape fixedTimeAuxInputShape = auxInputShape; in executeTyped()
232 fixedTimeAuxInputShape = removeFirstDim(auxInputShape); in executeTyped()
|
| D | RNN.cpp | 141 const Shape& auxInputShape, const T* hiddenStateInputData, const T* biasData, in RNNStep() argument
159 aux_input_size = auxInputShape.dimensions[1]; in RNNStep()
234 const _Float16* auxInputData, const Shape& auxInputShape,
250 const float* auxInputData, const Shape& auxInputShape,
|
| D | RNN.h | 57 const Shape& auxInputShape, const T* hiddenStateInputData,
|