Searched refs:recurrent_to_output_weights_shape (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | LSTM.h | 116 const Shape& recurrent_to_output_weights_shape,
143 const Shape& recurrent_to_output_weights_shape,
169 const Shape& recurrent_to_output_weights_shape,
|
| D | LSTM.cpp | 370 const Shape& recurrent_to_output_weights_shape, const float* cell_to_input_weights_buffer, in LSTMEvalFloat32() argument
395 const uint32_t outputSize = getSizeOfDimension(recurrent_to_output_weights_shape, 1); in LSTMEvalFloat32()
446 recurrent_to_output_weights_shape, cell_to_input_weights_buffer, in LSTMEvalFloat32()
488 const Shape& recurrent_to_output_weights_shape, in LSTMEvalFloat16() argument
515 const uint32_t outputSize = getSizeOfDimension(recurrent_to_output_weights_shape, 1); in LSTMEvalFloat16()
685 recurrent_to_output_weights_float32.data(), recurrent_to_output_weights_shape, in LSTMEvalFloat16()
730 const Shape& recurrent_to_output_weights_shape, const float* cell_to_input_weights_buffer, in LSTMStep() argument
749 const uint32_t n_output = recurrent_to_output_weights_shape.dimensions[1]; in LSTMStep()
|