Home
last modified time | relevance | path

Searched refs:n_aux_input (Results 1 – 2 of 2) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceLSTM.cpp341 uint32_t n_aux_input = SizeOfDimension(input_, 2); in Prepare() local
347 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_input_weights_, 1), n_aux_input); in Prepare()
351 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_forget_weights_, 1), n_aux_input); in Prepare()
355 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_cell_weights_, 1), n_aux_input); in Prepare()
359 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_output_weights_, 1), n_aux_input); in Prepare()
366 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_input_weights_, 1), n_aux_input); in Prepare()
370 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_forget_weights_, 1), n_aux_input); in Prepare()
374 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_cell_weights_, 1), n_aux_input); in Prepare()
378 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_output_weights_, 1), n_aux_input); in Prepare()
DLSTM.cpp793 const uint32_t n_aux_input = aux_input_buffer == nullptr ? 0 : n_input; in LSTMStep() local
852 aux_input_to_input_weights_buffer, n_cell, n_aux_input, aux_input_buffer, in LSTMStep()
857 aux_input_to_forget_weights_buffer, n_cell, n_aux_input, aux_input_buffer, n_batch, in LSTMStep()
860 aux_input_to_cell_weights_buffer, n_cell, n_aux_input, aux_input_buffer, n_batch, in LSTMStep()
863 aux_input_to_output_weights_buffer, n_cell, n_aux_input, aux_input_buffer, n_batch, in LSTMStep()