Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/keras/saving/
Dhdf5_format.py505 def transpose_input(from_cudnn): argument
525 order = 'F' if from_cudnn else 'C'
551 def convert_lstm_weights(weights, from_cudnn=True): argument
563 kernels = transform_kernels(weights[0], transpose_input(from_cudnn),
566 if from_cudnn:
576 weights = convert_lstm_weights(weights, from_cudnn=source == 'CuDNNLSTM')
588 def convert_gru_weights(weights, from_cudnn=True): argument
599 kernels = transform_kernels(weights[0], transpose_input(from_cudnn),
602 biases = np.array(weights[2]).reshape((2, -1) if from_cudnn else -1)
627 weights = convert_gru_weights(weights, from_cudnn=True)
[all …]