Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
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()