Home
last modified time | relevance | path

Searched refs:n_gates (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/keras/saving/
Dhdf5_format.py492 def transform_kernels(kernels, func, n_gates): argument
503 return np.hstack([func(k) for k in np.hsplit(kernels, n_gates)])
542 n_gates = 4
544 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 = 3
600 n_gates)
601 recurrent_kernels = transform_kernels(weights[1], lambda k: k.T, n_gates)
[all …]