message: The sharding test was missing the compute sets for the backward pass on IPU 1
  Summary: The backward pass should contain convolutions and updates for both IPUs
  Test Plan: test now ok
  Reviewers: georgep
diff: @@ -363,7 +363,10 @@ class BatchNormGraphCachingTest(test_util.TensorFlowTestCase):
  'GradientDescent/update_vs/batch_normalization_2/',
  'gradients/vs/conv3/Conv2D_grad/Conv2DBackpropInput/convolution.*.clone/WeightTranspose',
  'gradients/vs/conv3/Conv2D_grad/Conv2DBackpropFilter/convolution.*.clone/Conv_4x4',
  - 'GradientDescent/update_vs/conv3/kernel/ResourceApplyGradientDescent/subtract.*.clone/AddTo']
  + 'GradientDescent/update_vs/conv3/kernel/ResourceApplyGradientDescent/subtract.*.clone/AddTo',
  + 'gradients/vs/conv2/Conv2D_grad/Conv2DBackpropInput/convolution.89.clone/WeightTranspose',
  + 'gradients/vs/conv2/Conv2D_grad/Conv2DBackpropFilter/convolution.87.clone/Conv_4x4',
  + 'GradientDescent/update_vs/conv2/kernel/ResourceApplyGradientDescent/subtract.93.clone/AddTo']
  self.assertTrue(tu.check_all_compute_sets_and_list(cs_list, ok))
  