Home
last modified time | relevance | path

Searched refs:kHiddenStateOutTensor (Results 1 – 3 of 3) sorted by relevance

/frameworks/ml/nn/common/operations/
DRNN.h46 static constexpr int kHiddenStateOutTensor = 0; variable
DRNN.cpp45 hidden_state_out_ = GetOutput(operation, operands, kHiddenStateOutTensor); in RNN()
/frameworks/ml/nn/common/
DCpuExecutor.cpp1130 RunTimeOperandInfo& hiddenStateOut = operands[outs[RNN::kHiddenStateOutTensor]]; in executeOperation()