"forward" method. This graph will be preprocessed to inline all
      function and method calls. See Interpreting Graphs for details.
ipu(device=None)
  Moves all model parameters and buffers to the IPU.

  This also makes associated parameters and buffers different
  objects. So it should be called before constructing optimizer if
  the module will live on IPU while being optimized.

  Note:

    This method modifies the module in-place.

  Parameters:
     **device** (*int**, **optional*) -- if specified, all
     parameters will be copied to that device

  Returns:
     self

  Return type:
     Module

load_state_dict(state_dict, strict=True)
  Copies parameters and buffers from "state_dict" into this module
  and its descendants. If "strict" is "True", then the keys of
  "state_dict" must exactly match the keys returned by this
  module's "state_dict()" function.

  Parameters:
