Searched refs:from_cudnn (Results 1 – 1 of 1) sorted by relevance
455 def transpose_input(from_cudnn): argument475 order = 'F' if from_cudnn else 'C'501 def convert_lstm_weights(weights, from_cudnn=True): argument513 kernels = transform_kernels(weights[0], transpose_input(from_cudnn),516 if from_cudnn:526 weights = convert_lstm_weights(weights, from_cudnn=source == 'CuDNNLSTM')538 def convert_gru_weights(weights, from_cudnn=True): argument549 kernels = transform_kernels(weights[0], transpose_input(from_cudnn),552 biases = np.array(weights[2]).reshape((2, -1) if from_cudnn else -1)577 weights = convert_gru_weights(weights, from_cudnn=True)[all …]