Searched refs:get_memory_usage (Results 1 – 5 of 5) sorted by relevance
48 name: "get_memory_usage"
52 name: "get_memory_usage"
627 def get_memory_usage(device): function
608 self.assertEqual(config.get_memory_usage(device), info['current'])625 config.get_memory_usage('CPU:0')632 config.get_memory_usage('unknown_device:0')
3865 `tf.config.experimental.get_memory_usage` in favor of this new function.4639 * Adds `tf.config.experimental.get_memory_usage` to return total memory usage