Searched refs:get_memory_usage (Results 1 – 7 of 7) sorted by relevance
604 self.assertEqual(config.get_memory_usage(device), info['current'])617 config.get_memory_usage('CPU:0')624 config.get_memory_usage('unknown_device')646 config.get_memory_usage('GPU')
572 def get_memory_usage(device): function
33 def get_memory_usage(system_facade): function
185 mem_usage = '%.3f' % enterprise_longevity_helper.get_memory_usage(
48 name: "get_memory_usage"
113 `tf.config.experimental.get_memory_usage` in favor of this new function.501 * Adds `tf.config.experimental.get_memory_usage` to return total memory usage of the device.