Searched refs:input_to_output_weights_ (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | LSTM.cpp | 61 input_to_output_weights_ = GetInput(operation, operands, kInputToOutputWeightsTensor); in LSTMCell()
351 const uint32_t n_cell = SizeOfDimension(input_to_output_weights_, 0); in Prepare()
352 NN_CHECK_EQ(NumDimensions(input_to_output_weights_), 2u); in Prepare()
353 NN_CHECK_EQ(SizeOfDimension(input_to_output_weights_, 1), n_input); in Prepare()
362 input_to_output_weights_, recurrent_to_input_weights_, recurrent_to_forget_weights_, in Prepare()
999 GetBuffer<const float>(input_to_output_weights_), in Eval()
1000 input_to_output_weights_->shape(), in Eval()
1034 GetBuffer<const _Float16>(input_to_output_weights_), in Eval()
1035 input_to_output_weights_->shape(), in Eval()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | LSTM.h | 211 const RunTimeOperandInfo* input_to_output_weights_; variable
|