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.cpp107 *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() local
143 NN_RET_CHECK(getLinkingMode(context, &linkingMode)); in executeTyped()
144 if (linkingMode == LinkingMode::CROSS_LINKING) { in executeTyped()
234 if (linkingMode == LinkingMode::PARALLEL_LINKING) { in executeTyped()
381 LinkingMode linkingMode; in prepare() local
382 NN_RET_CHECK(getLinkingMode(context, &linkingMode)); in prepare()
410 if (linkingMode != LinkingMode::PARALLEL_LINKING) { in prepare()
[all …]
H A DBidirectionalSequenceLSTM.cpp57 bool getLinkingMode(bool hasAuxInput, bool hasAuxWeights, LinkingMode* linkingMode) { in getLinkingMode() argument
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() local
332 NN_RET_CHECK(getLinkingMode(has_aux_input, has_fw_aux_weights, &linkingMode)); in Prepare()
392 linkingMode == LinkingMode::PARALLEL_LINKING ? aux_input_ : input_; in Prepare()
453 LinkingMode linkingMode; in Eval() local
454 NN_RET_CHECK(getLinkingMode(has_aux_input, has_aux_weights, &linkingMode)); in Eval()
461 if (linkingMode == LinkingMode::PARALLEL_LINKING) { in Eval()
[all …]