Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DBidirectionalSequenceRNN.cpp89 enum class LinkingMode { enum
107 *linkingMode = LinkingMode::NO_LINKING; in getLinkingMode()
109 *linkingMode = LinkingMode::PARALLEL_LINKING; in getLinkingMode()
111 *linkingMode = LinkingMode::CROSS_LINKING; in getLinkingMode()
142 LinkingMode linkingMode; in executeTyped()
144 if (linkingMode == LinkingMode::CROSS_LINKING) { in executeTyped()
148 } else if (linkingMode == LinkingMode::PARALLEL_LINKING) { in executeTyped()
234 if (linkingMode == LinkingMode::PARALLEL_LINKING) { in executeTyped()
381 LinkingMode linkingMode; in prepare()
410 if (linkingMode != LinkingMode::PARALLEL_LINKING) { in prepare()
[all …]
H A DBidirectionalSequenceLSTM.cpp51 enum class LinkingMode { enum
57 bool getLinkingMode(bool hasAuxInput, bool hasAuxWeights, LinkingMode* linkingMode) { in getLinkingMode()
65 *linkingMode = LinkingMode::NO_LINKING; in getLinkingMode()
67 *linkingMode = LinkingMode::PARALLEL_LINKING; in getLinkingMode()
69 *linkingMode = LinkingMode::CROSS_LINKING; in getLinkingMode()
331 LinkingMode linkingMode; in Prepare()
392 linkingMode == LinkingMode::PARALLEL_LINKING ? aux_input_ : input_; in Prepare()
453 LinkingMode linkingMode; in Eval()
461 if (linkingMode == LinkingMode::PARALLEL_LINKING) { in Eval()
584 if (linkingMode == LinkingMode::PARALLEL_LINKING) { in Eval()