Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceLSTM.cpp342 int n_aux_input = SizeOfDimension(input_, 2); in Prepare() local
348 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_input_weights_, 1), n_aux_input); in Prepare()
352 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_forget_weights_, 1), n_aux_input); in Prepare()
356 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_cell_weights_, 1), n_aux_input); in Prepare()
360 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_output_weights_, 1), n_aux_input); in Prepare()
367 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_input_weights_, 1), n_aux_input); in Prepare()
371 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_forget_weights_, 1), n_aux_input); in Prepare()
375 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_cell_weights_, 1), n_aux_input); in Prepare()
379 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
853 aux_input_to_input_weights_buffer, n_cell, n_aux_input, aux_input_buffer, in LSTMStep()
859 aux_input_to_forget_weights_buffer, n_cell, n_aux_input, aux_input_buffer, n_batch, in LSTMStep()
862 aux_input_to_cell_weights_buffer, n_cell, n_aux_input, aux_input_buffer, n_batch, in LSTMStep()
865 aux_input_to_output_weights_buffer, n_cell, n_aux_input, aux_input_buffer, n_batch, in LSTMStep()