(w)
| 1024 | The converted weights that can be feed to CuDNN ops as param. |
| 1025 | """ |
| 1026 | def convert(w): |
| 1027 | return array_ops.transpose(w) if transpose_weights else w |
| 1028 | |
| 1029 | weights = [array_ops.reshape(convert(x), shape) for x in weights] |
| 1030 | biases = [array_ops.reshape(x, shape) for x in biases] |