Searched refs:fixedTimeInputShape (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceRNN.cpp | 90 Shape fixedTimeInputShape = inputShape; in executeTyped() local
91 fixedTimeInputShape.dimensions.resize(2); in executeTyped()
92 fixedTimeInputShape.dimensions[0] = inputShape.dimensions[1]; in executeTyped()
93 fixedTimeInputShape.dimensions[1] = inputShape.dimensions[2]; in executeTyped()
96 RNN::RNNStep<T>(input, fixedTimeInputShape, hiddenState, bias, weights, weightsShape, in executeTyped()
|
| D | BidirectionalSequenceRNN.cpp | 198 Shape fixedTimeInputShape = removeFirstDim(inputShape); in executeTyped() local
236 RNN::RNNStep<T>(inputBatchPtr, fixedTimeInputShape, auxInputBatchPtr, in executeTyped()
264 RNN::RNNStep<T>(inputBatchPtr, fixedTimeInputShape, auxInputBatchPtr, in executeTyped()
|