code: def test_mkl_simple_rewrite(self, gen):
      cpu_model, (shape,) = gen()
      cpu_model = deterministic_io(cpu_model)
      mkl_model = rewrite_graph.rewrite_model_helper_simple(cpu_model)
      X = np.random.randn(*shape).astype(np.float32)
  
      def run(model):
          self.ws.run(model.InitProto())
          self.ws.create_blob(model.Proto().external_input[0]).feed(X)
          self.ws.run(model.Proto())
          return self.ws.blobs[model.Proto().external_output[0]].fetch()
  
      np.testing.assert_allclose(
          run(cpu_model), run(mkl_model), atol=1e-4, rtol=1e-4
      )
  
imports: import unittest
  import numpy
  import copy
  from hypothesis import given
  import hypothesis.strategies
  from caffe2.python.model_helper import ModelHelper
  from caffe2.python.models import resnet
  from caffe2.python import workspace, brew
  import caffe2.python.hypothesis_test_util
  import caffe2.python.mkl.rewrite_graph
label_desc: Use image node_id 1 for calling the MKLRewriteTest obj's underlying member method code with example usage: obj.test_mkl_simple_rewrite(gen) and returns: self