code: def __call__(self, x):
      return tf.reshape(tf.add(tf.matmul(x, self.w), self.b), [-1])
  
file_path: mlflow/tests/tensorflow/test_tensorflow2_core_model_export.py
imports: import collections
  import os
  from unittest import mock
  import numpy
  import pytest
  import tensorflow
  import mlflow.tensorflow
  from mlflow.models import Model, infer_signature
label_desc: Use image node_id 2 for calling the ToyModel obj's underlying member method code with example usage: obj.__call__(x) and returns: tf