def get_video_shape(params, is_space_to_depth=False):
  """Returns exact video shape as model's input."""
  if is_space_to_depth:
    video_shape = [params.num_frames // 2,
                   params.frame_size // 2,
                   params.frame_size // 2,
                   24]
  else:
    video_shape = [params.num_frames,
                   params.frame_size,
                   params.frame_size,
                   3]
  return video_shape