code: def testResetMemoryStats(self, device_type):
      device = f"{device_type}:0"
      with ops.device(device):
          x = array_ops.zeros((1000, 1000), dtype=dtypes.float32)
      config.reset_memory_stats(device)
      info1 = config.get_memory_info(device)
      self.assertGreaterEqual(info1["peak"], 4 * 1000 * 1000)
      self.assertGreaterEqual(info1["peak"], info1["current"])
      self.assertGreater(info1["current"], 0)
  
      del x  # With CPython, causes tensor memory to be immediately freed
      config.reset_memory_stats(device)
      info2 = config.get_memory_info(device)
      self.assertLess(info2["peak"], info1["peak"])
  
imports: from absl.testing import parameterized
  from tensorflow.core.protobuf import cluster_pb2
  from tensorflow.core.protobuf import config_pb2
  from tensorflow.core.protobuf import rewriter_config_pb2
  from tensorflow.core.protobuf import tensorflow_server_pb2
  from tensorflow.python.eager import context
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import config
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import test
  from tensorflow.python.util import compat
label_desc: Use image node_id 15 for calling the DeviceTest obj's underlying member method code with example usage: obj.testResetMemoryStats(device_type) without return types