Home
last modified time | relevance | path

Searched refs:kMergeOutputsParam (Results 1 – 5 of 5) sorted by relevance

/packages/modules/NeuralNetworks/common/types/operations/include/
DBidirectionalSequenceRNN.h43 constexpr uint32_t kMergeOutputsParam = 14; variable
DBidirectionalSequenceLSTM.h132 static constexpr int kMergeOutputsParam = 51; variable
/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceRNN.cpp131 const bool mergeOutputs = context->getInputValue<bool>(kMergeOutputsParam); in executeTyped()
289 const bool mergeOutputs = context->getInputValue<bool>(kMergeOutputsParam); in prepare()
301 kBwHiddenStateTensor, kActivationParam, kTimeMajorParam, kMergeOutputsParam, in prepare()
DBidirectionalSequenceLSTM.cpp185 const auto& mergeOutputsOperand = *GetInput(operation, operands, kMergeOutputsParam); in BidirectionalSequenceLSTM()
240 kMergeOutputsParam, in Prepare()
260 const auto& mergeOutputsOperand = *GetInput(operation, operands, kMergeOutputsParam); in Prepare()
/packages/modules/NeuralNetworks/common/
DCpuExecutor.cpp1000 operands[ins[BidirectionalSequenceLSTM::kMergeOutputsParam]]); in executeOperation()