Searched refs:get_memory_info (Results 1 – 6 of 6) sorted by relevance
527 def get_memory_info(device): function565 return context.context().get_memory_info(device)604 return get_memory_info(device)['current']
600 info = config.get_memory_info(device)609 info = config.get_memory_info('GPU:0')615 config.get_memory_info('CPU:0')622 config.get_memory_info('unknown_device')630 peak1 = config.get_memory_info('GPU:0')['peak']633 peak2 = config.get_memory_info('GPU:0')['peak']636 peak3 = config.get_memory_info('GPU:0')['peak']638 self.assertGreaterEqual(peak3, config.get_memory_info('GPU:0')['current'])
44 name: "get_memory_info"
1441 def get_memory_info(self, dev): member in Context1450 return self.get_memory_info(dev)["current"]
111 * Added `tf.config.experimental.get_memory_info`, returning a dict with the