Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceRNN.cpp97 bool getLinkingMode(IOperationExecutionContext* context, LinkingMode* linkingMode) { in getLinkingMode() argument
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() local
145 NN_RET_CHECK(getLinkingMode(context, &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.cpp58 bool getLinkingMode(bool hasAuxInput, bool hasAuxWeights, LinkingMode* linkingMode) { in getLinkingMode() argument
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() local
333 NN_RET_CHECK(getLinkingMode(has_aux_input, has_fw_aux_weights, &linkingMode)); in Prepare()
393 linkingMode == LinkingMode::PARALLEL_LINKING ? aux_input_ : input_; in Prepare()
454 LinkingMode linkingMode; in Eval() local
455 NN_RET_CHECK(getLinkingMode(has_aux_input, has_aux_weights, &linkingMode)); in Eval()
462 if (linkingMode == LinkingMode::PARALLEL_LINKING) { in Eval()
[all …]