Home
last modified time | relevance | path

Searched refs:hasAuxInput (Results 1 – 2 of 2) sorted by relevance

/frameworks/ml/nn/common/operations/
DRNN.cpp157 bool hasAuxInput = (auxInputData != nullptr); in RNNStep() local
158 if (hasAuxInput) { in RNNStep()
169 if (hasAuxInput) { in RNNStep()
178 if (hasAuxInput) { in RNNStep()
195 if (hasAuxInput) { in RNNStep()
DLSTM.cpp403 const bool hasAuxInput = (aux_input_buffer != nullptr); in LSTMEvalFloat32() local
411 if (hasAuxInput) { in LSTMEvalFloat32()
423 hasAuxInput ? (timeMajor ? aux_input_buffer : transposedAuxInput.data()) : nullptr; in LSTMEvalFloat32()
433 hasAuxInput ? (auxInputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1))) in LSTMEvalFloat32()
459 if (hasAuxInput) { in LSTMEvalFloat32()
638 const bool hasAuxInput = (aux_input_buffer != nullptr); in LSTMEvalFloat16() local
647 if (hasAuxInput) { in LSTMEvalFloat16()
659 hasAuxInput ? (timeMajor ? aux_input_float32.data() : transposedAuxInput.data()) in LSTMEvalFloat16()
671 hasAuxInput ? (auxInputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1))) in LSTMEvalFloat16()
701 if (hasAuxInput) { in LSTMEvalFloat16()