Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceRNN.cpp50 constexpr uint32_t kMergeOutputsParam = 14; variable
162 const bool mergeOutputs = context->getInputValue<bool>(kMergeOutputsParam); in executeTyped()
349 const bool mergeOutputs = context->getInputValue<bool>(kMergeOutputsParam); in prepare()
361 kBwHiddenStateTensor, kActivationParam, kTimeMajorParam, kMergeOutputsParam, in prepare()
DBidirectionalSequenceLSTM.h135 static constexpr int kMergeOutputsParam = 51; variable
DBidirectionalSequenceLSTM.cpp186 const auto& mergeOutputsOperand = *GetInput(operation, operands, kMergeOutputsParam); in BidirectionalSequenceLSTM()
241 kMergeOutputsParam, in Prepare()
261 const auto& mergeOutputsOperand = *GetInput(operation, operands, kMergeOutputsParam); in Prepare()
/frameworks/ml/nn/common/
DCpuExecutor.cpp1055 operands[ins[BidirectionalSequenceLSTM::kMergeOutputsParam]]); in executeOperation()