Searched refs:n_fw_cell (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceLSTM.cpp | 273 const uint32_t n_fw_cell = SizeOfDimension(fw_input_to_output_weights_, 0); in Prepare() local
278 NN_CHECK_EQ(SizeOfDimension(fw_recurrent_to_output_weights_, 0), n_fw_cell); in Prepare()
297 fw_output_layer_norm_weights_, n_fw_input, n_fw_output, n_fw_cell, ¶ms_)) { in Prepare()
347 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_input_weights_, 0), n_fw_cell); in Prepare()
351 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_forget_weights_, 0), n_fw_cell); in Prepare()
355 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_cell_weights_, 0), n_fw_cell); in Prepare()
359 NN_RET_CHECK_EQ(SizeOfDimension(fw_aux_input_to_output_weights_, 0), n_fw_cell); in Prepare()
427 fw_scratch_shape_.dimensions = {n_batch, n_fw_cell * 3}; in Prepare()
430 fw_scratch_shape_.dimensions = {n_batch, n_fw_cell * 4}; in Prepare()
|