Searched refs:fixedTimeAuxInputShape (Results 1 – 1 of 1) sorted by relevance
/frameworks/ml/nn/common/operations/ |
D | BidirectionalSequenceRNN.cpp | 179 Shape fixedTimeAuxInputShape = auxInputShape; in executeTyped() local 181 fixedTimeAuxInputShape = removeFirstDim(auxInputShape); in executeTyped() 197 fixedTimeAuxInputShape, fwHiddenState, fwBias, fwWeights, fwWeightsShape, in executeTyped() 226 fixedTimeAuxInputShape, bwHiddenState, bwBias, bwWeights, bwWeightsShape, in executeTyped()
|