Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DLSTM.h216 const RunTimeOperandInfo* recurrent_to_cell_weights_; variable
H A DLSTM.cpp66 recurrent_to_cell_weights_ = GetInput(operation, operands, kRecurrentToCellWeightsTensor); in LSTMCell()
363 recurrent_to_cell_weights_, recurrent_to_output_weights_, cell_to_input_weights_, in Prepare()
1001 GetBuffer<const float>(recurrent_to_cell_weights_), in Eval()
1036 GetBuffer<const _Float16>(recurrent_to_cell_weights_), in Eval()