code: def forward(self, x):
      if ~self.mode:
          return x * x + self.param[0] * x + self.param[1]
      else:
          return_val = 10 * x * x * x
          return_val += self.param[0] * x * x
          return_val += self.param[1] * x + self.param[2]
          return return_val
  
file_path: ray/python/ray/train/examples/horovod/horovod_tune_example.py
imports: import time
  import numpy
  import torch
  import ray
  import ray.train.torch
  from ray import train, tune
  from ray.train import ScalingConfig
  from ray.train.horovod import HorovodTrainer
  from ray.tune.tune_config import TuneConfig
  from ray.tune.tuner import Tuner
label_desc: Use image node_id 2 for calling the Net obj's underlying member method code with example usage: obj.forward(x) and returns: unknown, return_val