Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceRNN.cpp45 constexpr uint32_t kMergeOutputsParam = 14; variable
111 int32_t mergeOutputs = context->getInputValue<bool>(kMergeOutputsParam); in executeTyped()
273 int32_t mergeOutputs = context->getInputValue<bool>(kMergeOutputsParam); in prepare()
284 kBwHiddenStateTensor, kActivationParam, kTimeMajorParam, kMergeOutputsParam, in prepare()
DBidirectionalSequenceLSTM.h133 static constexpr int kMergeOutputsParam = 51; variable
DBidirectionalSequenceLSTM.cpp150 params_.merge_outputs = getScalarData<bool>(*GetInput(operation, operands, kMergeOutputsParam)); in BidirectionalSequenceLSTM()
/frameworks/ml/nn/common/
DCpuExecutor.cpp1102 mOperands[ins[BidirectionalSequenceLSTM::kMergeOutputsParam]]); in executeOperation()