message: Fix v2 test
  Summary: Ref T25867
  Test Plan: Test passes
  Reviewers: #tensorflow, hakons, #framework_ip_review_-_any_oss_or_third-party_code_use_has_been_approved
  Maniphest Tasks: T25867
diff: @@ -400,6 +400,8 @@ class PipeliningConvClassifyTest(test_util.TensorFlowTestCase):
  x = array_ops.placeholder(np.float32, shape=[1, 224])
  l = array_ops.placeholder(np.int32, shape=[1])
  + with tu.ipu_session() as sess:
  +
  with ops.device("/device:IPU:0"):
  compiled_model_pipeline = ipu_compiler.compile(model_pipeline,
  inputs=[x, l])
  @@ -407,8 +409,6 @@ class PipeliningConvClassifyTest(test_util.TensorFlowTestCase):
  tu.move_variable_initialization_to_cpu()
  outfeed_queue.dequeue()
  - with tu.ipu_session() as sess:
  -
  report = tu.ReportJSON(self, sess, pipelining=True)
  sess.run(variables.global_variables_initializer())
  report.reset()
  