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 | 413 const Shape& recurrent_to_output_weights_shape, const float* cell_to_input_weights_buffer, in LSTMEvalFloat32() argument
438 const uint32_t outputSize = getSizeOfDimension(recurrent_to_output_weights_shape, 1); in LSTMEvalFloat32()
489 recurrent_to_output_weights_shape, cell_to_input_weights_buffer, in LSTMEvalFloat32()
531 const Shape& recurrent_to_output_weights_shape, in LSTMEvalFloat16() argument
558 const uint32_t outputSize = getSizeOfDimension(recurrent_to_output_weights_shape, 1); in LSTMEvalFloat16()
728 recurrent_to_output_weights_float32.data(), recurrent_to_output_weights_shape, in LSTMEvalFloat16()
773 const Shape& recurrent_to_output_weights_shape, const float* cell_to_input_weights_buffer, in LSTMStep() argument
792 const uint32_t n_output = recurrent_to_output_weights_shape.dimensions[1]; in LSTMStep()
|