Searched refs:n_gates (Results 1 – 1 of 1) sorted by relevance
442 def transform_kernels(kernels, func, n_gates): argument453 return np.hstack([func(k) for k in np.hsplit(kernels, n_gates)])492 n_gates = 4494 if bias_shape == (2 * units * n_gates,):496 elif bias_shape == (units * n_gates,):514 n_gates)515 recurrent_kernels = transform_kernels(weights[1], lambda k: k.T, n_gates)536 n_gates = 3550 n_gates)551 recurrent_kernels = transform_kernels(weights[1], lambda k: k.T, n_gates)[all …]