Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DLSTM.cpp409 const Shape& input_to_output_weights_shape, const float* recurrent_to_input_weights_buffer, in LSTMEvalFloat32()
526 const _Float16* input_to_output_weights_buffer, const Shape& input_to_output_weights_shape, in LSTMEvalFloat16()
769 const Shape& input_to_output_weights_shape, const float* recurrent_to_input_weights_buffer, in LSTMStep()