Converting models in Tensorflow 1.6

I'm using tensorflow 1.6 due to newer version of CUDA. After changing few imports the code is working fine, except transferring trained models to cpu. There is no RNNParamsSaveable class in 1.6 version, so I changed it to CudnnGRUSaveable with another parameters:
`params_saveable = cudnn_rnn_ops.CudnnGRUSaveable(fw_params, 1, dim, 400, direction="bidirectional")`
But I'm getting an error:
`2018-05-22 13:45:03.151296: F tensorflow/contrib/cudnn_rnn/kernels/cudnn_rnn_ops.cc:203] Check failed: offset + size <= device_memory.size() The slice is not within the region of DeviceMemory.`
I have GeForce GTX 1080 Ti, 12 Gb, so probably I have enough memory, but still doing something wrong.
Just wondering maybe somebody knows how to resolve this problem.