Searched refs:n_gates (Results 1 – 1 of 1) sorted by relevance
492 def transform_kernels(kernels, func, n_gates): argument503 return np.hstack([func(k) for k in np.hsplit(kernels, n_gates)])542 n_gates = 4544 if bias_shape == (2 * units * n_gates,):546 elif bias_shape == (units * n_gates,):564 n_gates)565 recurrent_kernels = transform_kernels(weights[1], lambda k: k.T, n_gates)586 n_gates = 3600 n_gates)601 recurrent_kernels = transform_kernels(weights[1], lambda k: k.T, n_gates)[all …]