Searched refs:recurrent_to_output_weights_ (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | LSTM.cpp | 67 recurrent_to_output_weights_ = GetInput(operation, operands, kRecurrentToOutputWeightsTensor); in LSTMCell()
355 NN_CHECK_EQ(NumDimensions(recurrent_to_output_weights_), 2u); in Prepare()
356 NN_CHECK_EQ(SizeOfDimension(recurrent_to_output_weights_, 0), n_cell); in Prepare()
357 const uint32_t n_output = SizeOfDimension(recurrent_to_output_weights_, 1); in Prepare()
363 recurrent_to_cell_weights_, recurrent_to_output_weights_, cell_to_input_weights_, in Prepare()
1004 GetBuffer<const float>(recurrent_to_output_weights_), in Eval()
1005 recurrent_to_output_weights_->shape(), in Eval()
1039 GetBuffer<const _Float16>(recurrent_to_output_weights_), in Eval()
1040 recurrent_to_output_weights_->shape(), in Eval()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | LSTM.h | 216 const RunTimeOperandInfo* recurrent_to_output_weights_; variable
|