Searched refs:from_cudnn (Results 1 – 1 of 1) sorted by relevance
505 def transpose_input(from_cudnn): argument525 order = 'F' if from_cudnn else 'C'551 def convert_lstm_weights(weights, from_cudnn=True): argument563 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): argument599 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 …]