Searched refs:n_bw_cell (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BidirectionalSequenceLSTM.cpp | 280 const uint32_t n_bw_cell = SizeOfDimension(bw_input_to_output_weights_, 0); in Prepare() local
283 NN_CHECK_EQ(SizeOfDimension(bw_recurrent_to_output_weights_, 0), n_bw_cell); in Prepare()
365 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_input_weights_, 0), n_bw_cell); in Prepare()
369 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_forget_weights_, 0), n_bw_cell); in Prepare()
373 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_cell_weights_, 0), n_bw_cell); in Prepare()
377 NN_RET_CHECK_EQ(SizeOfDimension(bw_aux_input_to_output_weights_, 0), n_bw_cell); in Prepare()
404 bw_output_layer_norm_weights_, n_bw_input, n_bw_output, n_bw_cell, ¶ms_)) { in Prepare()
427 bw_scratch_shape_.dimensions = {n_batch, n_bw_cell * 3}; in Prepare()
430 bw_scratch_shape_.dimensions = {n_batch, n_bw_cell * 4}; in Prepare()
|