Home
last modified time | relevance | path

Searched refs:cell_state_out_ (Results 1 – 2 of 2) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DLSTM.h240 RunTimeOperandInfo* cell_state_out_; variable
H A DLSTM.cpp125 cell_state_out_ = GetOutput(operation, operands, kCellStateOutTensor); in LSTMCell()
1024 GetBuffer<float>(output_state_out_), GetBuffer<float>(cell_state_out_), in Eval()
1060 GetBuffer<_Float16>(cell_state_out_), GetBuffer<_Float16>(output_), in Eval()