Searched refs:attn_bw_cell (Results 1 – 1 of 1) sorted by relevance
358 def attn_bw_cell(): function367 attn_bw_cell = bw_cell373 [attn_bw_cell() for _ in range(num_layers)], state_is_tuple=False)