Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceRNN.cpp91 enum class LinkingMode { enum
97 bool getLinkingMode(IOperationExecutionContext* context, LinkingMode* linkingMode) { in getLinkingMode()
109 *linkingMode = LinkingMode::NO_LINKING; in getLinkingMode()
111 *linkingMode = LinkingMode::PARALLEL_LINKING; in getLinkingMode()
113 *linkingMode = LinkingMode::CROSS_LINKING; in getLinkingMode()
144 LinkingMode linkingMode; in executeTyped()
146 if (linkingMode == LinkingMode::CROSS_LINKING) { in executeTyped()
150 } else if (linkingMode == LinkingMode::PARALLEL_LINKING) { in executeTyped()
153 const bool hasAuxInput = (linkingMode == LinkingMode::CROSS_LINKING || in executeTyped()
154 linkingMode == LinkingMode::PARALLEL_LINKING); in executeTyped()
[all …]
DBidirectionalSequenceLSTM.cpp52 enum class LinkingMode { enum
58 bool getLinkingMode(bool hasAuxInput, bool hasAuxWeights, LinkingMode* linkingMode) { in getLinkingMode()
66 *linkingMode = LinkingMode::NO_LINKING; in getLinkingMode()
68 *linkingMode = LinkingMode::PARALLEL_LINKING; in getLinkingMode()
70 *linkingMode = LinkingMode::CROSS_LINKING; in getLinkingMode()
332 LinkingMode linkingMode; in Prepare()
393 linkingMode == LinkingMode::PARALLEL_LINKING ? aux_input_ : input_; in Prepare()
454 LinkingMode linkingMode; in Eval()
462 if (linkingMode == LinkingMode::PARALLEL_LINKING) { in Eval()
585 if (linkingMode == LinkingMode::PARALLEL_LINKING) { in Eval()