Searched refs:fixedTimeInputShape (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceRNN.cpp | 94 Shape fixedTimeInputShape = inputShape; in executeTyped() local
95 fixedTimeInputShape.dimensions.resize(2); in executeTyped()
96 fixedTimeInputShape.dimensions[0] = inputShape.dimensions[1]; in executeTyped()
97 fixedTimeInputShape.dimensions[1] = inputShape.dimensions[2]; in executeTyped()
100 RNN::RNNStep<T>(input, fixedTimeInputShape, hiddenState, bias, weights, weightsShape, in executeTyped()
|
| D | BidirectionalSequenceRNN.cpp | 178 Shape fixedTimeInputShape = removeFirstDim(inputShape); in executeTyped() local
196 RNN::RNNStep<T>(inputBatchPtr, fixedTimeInputShape, auxInputBatchPtr, in executeTyped()
225 RNN::RNNStep<T>(inputBatchPtr, fixedTimeInputShape, auxInputBatchPtr, in executeTyped()
|