Home
last modified time | relevance | path

Searched defs:inputRank (Results 1 – 3 of 3) sorted by relevance

/frameworks/ml/nn/common/operations/
DReduce.cpp50 const uint32_t inputRank = getNumberOfDimensions(inputShape); in compute() local
107 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare() local
DUnidirectionalSequenceLSTM.cpp185 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare() local
DLSTM.cpp386 const uint32_t inputRank = getNumberOfDimensions(input_shape); in LSTMEvalFloat32() local
506 const uint32_t inputRank = getNumberOfDimensions(input_shape); in LSTMEvalFloat16() local