Home
last modified time | relevance | path

Searched refs:kCellStateOutTensor (Results 1 – 6 of 6) sorted by relevance

/frameworks/ml/nn/common/operations/
DQuantizedLSTM.h48 static constexpr int kCellStateOutTensor = 0; variable
DLSTM.h102 static constexpr int kCellStateOutTensor = 2; variable
DQuantizedLSTMTest.cpp140 setOutputTensor(&execution, QuantizedLSTMCell::kCellStateOutTensor, &cellStateOut_), in invoke()
DQuantizedLSTM.cpp238 cellStateOut_ = GetOutput(operation, operands, kCellStateOutTensor); in QuantizedLSTMCell()
DLSTM.cpp116 cell_state_out_ = GetOutput(operation, operands, kCellStateOutTensor); in LSTMCell()
/frameworks/ml/nn/common/
DCpuExecutor.cpp1120 RunTimeOperandInfo& cellStateOut = mOperands[outs[LSTMCell::kCellStateOutTensor]]; in executeOperation()
1684 mOperands[outs[QuantizedLSTMCell::kCellStateOutTensor]]; in executeOperation()