class_docstr: Solver for raw tensorflow model.
class_docstr_tok: ["Solver","for","raw","tensorflow","model","."]
docstr: Get the input function for model export.
docstr_tok: ["Get","the","input","function","for","model","export","."]
code: def export_input(self, mode):
      """Get the input function for model export."""
      super().input_fn(mode)
      return self.task.export_inputs()
  
imports: import re
  import math
  import numpy
  import delta.compat
  from absl import logging
  from delta.utils.solver.base_solver import Solver
  from delta import utils
  from delta.utils.register import registers
  from delta.utils.solver.utils.solver_utils import get_checkpoint_dir
  from delta.utils.solver.utils.solver_utils import get_ckpt_state
  from delta.utils.solver.utils.solver_utils import get_session_conf
  from delta.utils.solver.utils.solver_utils import to_saved_model
  from delta.utils.solver.utils.solver_utils import run_metrics
label_desc: Use image node_id 4 for calling the RawSolver obj's underlying member method code with example usage: obj.export_input(mode) and returns: self