Searched refs:n_bw_output (Results 1 – 1 of 1) sorted by relevance
285 const uint32_t n_bw_output = SizeOfDimension(bw_recurrent_to_output_weights_, 1); in Prepare() local390 fwOutputShape->dimensions[2] = params_.merge_outputs ? n_fw_output + n_bw_output : n_fw_output; in Prepare()405 bw_output_layer_norm_weights_, n_bw_input, n_bw_output, n_bw_cell, ¶ms_)) { in Prepare()416 bwOutputShape->dimensions[2] = n_bw_output; in Prepare()442 const uint32_t n_bw_output = SizeOfDimension(bw_recurrent_to_output_weights_, 1); in Eval() local446 bw_output_dims[2] = n_bw_output; in Eval()