Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DLSTM.h219 const RunTimeOperandInfo* cell_to_input_weights_; variable
DLSTM.cpp69 cell_to_input_weights_ = GetInput(operation, operands, kCellToInputWeightsTensor); // optional in LSTMCell()
363 recurrent_to_cell_weights_, recurrent_to_output_weights_, cell_to_input_weights_, in Prepare()
1008 GetBuffer<const float>(cell_to_input_weights_), in Eval()
1043 GetOptionalBuffer<const _Float16>(cell_to_input_weights_), in Eval()