Searched refs:auxInputShape (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BidirectionalSequenceRNN.cpp | 125 Shape auxInputShape = context->getInputShape(kAuxInputTensor); in executeTyped() local
153 auxInputTransposed.resize(getNumberOfElements(auxInputShape)); in executeTyped()
163 transposeFirstTwoDims(auxInput, auxInputShape, auxInputTransposed.data()); in executeTyped()
180 std::swap(auxInputShape.dimensions[0], auxInputShape.dimensions[1]); in executeTyped()
193 auxInputSize = getSizeOfDimension(auxInputShape, 2); in executeTyped()
199 Shape fixedTimeAuxInputShape = auxInputShape; in executeTyped()
201 fixedTimeAuxInputShape = removeFirstDim(auxInputShape); in executeTyped()
|
| D | RNN.cpp | 137 const Shape& auxInputShape, const T* hiddenStateInputData, const T* biasData, in RNNStep() argument
155 aux_input_size = auxInputShape.dimensions[1]; in RNNStep()
230 const _Float16* auxInputData, const Shape& auxInputShape,
246 const float* auxInputData, const Shape& auxInputShape,
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | RNN.h | 63 const Shape& auxInputShape, const T* hiddenStateInputData,
|