Home
last modified time | relevance | path

Searched defs:cell_to_input_weights_buffer (Results 1 – 1 of 1) sorted by relevance

/frameworks/ml/nn/common/operations/
DLSTM.cpp413 const Shape& recurrent_to_output_weights_shape, const float* cell_to_input_weights_buffer, in LSTMEvalFloat32()
532 const _Float16* cell_to_input_weights_buffer, const _Float16* cell_to_forget_weights_buffer, in LSTMEvalFloat16()
773 const Shape& recurrent_to_output_weights_shape, const float* cell_to_input_weights_buffer, in LSTMStep()