Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceRNN.cpp66 bool getLinkingMode(IOperationExecutionContext* context, LinkingMode* linkingMode) { in getLinkingMode() argument
78 *linkingMode = LinkingMode::NO_LINKING; in getLinkingMode()
80 *linkingMode = LinkingMode::PARALLEL_LINKING; in getLinkingMode()
82 *linkingMode = LinkingMode::CROSS_LINKING; in getLinkingMode()
113 LinkingMode linkingMode; in executeTyped() local
114 NN_RET_CHECK(getLinkingMode(context, &linkingMode)); in executeTyped()
115 if (linkingMode == LinkingMode::CROSS_LINKING) { in executeTyped()
119 } else if (linkingMode == LinkingMode::PARALLEL_LINKING) { in executeTyped()
122 const bool hasAuxInput = (linkingMode == LinkingMode::CROSS_LINKING || in executeTyped()
123 linkingMode == LinkingMode::PARALLEL_LINKING); in executeTyped()
[all …]
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 …]