class_docstr: SV2P with specific changes for atari pipeline.
class_docstr_tok: ["SV2P","with","specific","changes","for","atari","pipeline","."]
code: def save_internal_states_ops(self, internal_states):
      ops = [
          [tf.assign(x[0], y[0]), tf.assign(x[1], y[1])]
          for x, y in zip(self.internal_states, internal_states)
      ]
      return ops
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from tensor2tensor.layers import common_layers
  from tensor2tensor.layers import common_video
  from tensor2tensor.layers import discretization
  from tensor2tensor.models.video import base
  from tensor2tensor.models.video import base_vae
  from tensor2tensor.utils import contrib
  from tensor2tensor.utils import registry
  import tensorflow.compat.v1
label_desc: Use image node_id 4 for calling the NextFrameSv2pAtari obj's underlying member method code with example usage: obj.save_internal_states_ops(internal_states) and returns: ops