message: Update microbenchmarks after cl/271457940
diff: @@ -679,8 +679,7 @@ class MicroBenchmarks(test.Benchmark):
  tangent = random_ops.random_uniform(shape).cpu()
  def func():
  - with forwardprop.ForwardGradientAccumulator() as acc:
  - acc.watch(m, tangent)
  + with forwardprop.ForwardGradientAccumulator(m, tangent) as acc:
  result = math_ops.matmul(m, m, transpose_b=True)
  return result, acc.jvp(result)
  @@ -693,8 +692,7 @@ class MicroBenchmarks(test.Benchmark):
  with ops.device(CPU):
  @def_function.function
  def compiled_function(x, tangent):
  - with forwardprop.ForwardGradientAccumulator() as acc:
  - acc.watch(x, tangent)
  + with forwardprop.ForwardGradientAccumulator(x, tangent) as acc:
  result = math_ops.matmul(x, x, transpose_b=True)
  return result, acc.jvp(result)
  @@ -713,8 +711,7 @@ class MicroBenchmarks(test.Benchmark):
  @def_function.function()
  def compiled_function(x, tangent):
  - with forwardprop.ForwardGradientAccumulator() as acc:
  - acc.watch(x, tangent)
  + with forwardprop.ForwardGradientAccumulator(x, tangent) as acc:
  result = matmul(x, x, transpose_b=True)
  return result, acc.jvp(result)
  @@ -734,8 +731,7 @@ class MicroBenchmarks(test.Benchmark):
  matmul = def_function.function(math_ops.matmul)
  def func():
  - with forwardprop.ForwardGradientAccumulator() as acc:
  - acc.watch(m, tangent)
  + with forwardprop.ForwardGradientAccumulator(m, tangent) as acc:
  result = matmul(m, m, transpose_b=True)
  return result, acc.jvp(result)
  