Searched refs:num_dir_layers (Results 1 – 1 of 1) sorted by relevance
887 int64_t num_dir_layers = rnn.num_directions() * rnn.num_layers; in get_parameters() local896 data_ptrs.reserve(num_dir_layers * (2 * 2 + 1)); in get_parameters()898 data_ptrs.reserve(num_dir_layers * 2 * 2); in get_parameters()900 for (const auto layer : c10::irange(num_dir_layers)) { in get_parameters()