Searched refs:get_memory_info (Results 1 – 9 of 9) sorted by relevance
133 self.assertGreater(context.context().get_memory_info('GPU:0')['current'], 0)142 context.context().get_memory_info('CPU:0')147 context.context().get_memory_info('unknown_device:0')152 context.context().get_memory_info('GPU')154 context.context().get_memory_info('GPU:')156 context.context().get_memory_info('GPU:CPU')
710 initial_usage = context.context().get_memory_info(714 final_usage = context.context().get_memory_info(736 initial_usage = context.context().get_memory_info(743 final_usage = context.context().get_memory_info(794 initial_usage = context.context().get_memory_info(800 final_usage = context.context().get_memory_info(
1592 def get_memory_info(self, dev): member in Context
604 info = config.get_memory_info(device)613 info = config.get_memory_info(f'{device_type}:0')623 config.get_memory_info('CPU:0')630 config.get_memory_info('unknown_device:0')637 context.context().get_memory_info('GPU')639 context.context().get_memory_info('GPU:')641 context.context().get_memory_info('GPU:CPU')649 peak1 = config.get_memory_info(device)['peak']653 peak2 = config.get_memory_info(device)['peak']656 peak3 = config.get_memory_info(device)['peak'][all …]
534 def get_memory_info(device): function576 return context.context().get_memory_info(device)659 return get_memory_info(device)['current']
67 base_memory = config.get_memory_info(device_name)["current"]95 config.get_memory_info(device_name)["peak"] - base_memory)100 config.get_memory_info(device_name)["peak"] - base_memory)
44 name: "get_memory_info"
48 name: "get_memory_info"
1809 peak memory returned by `tf.config.experimental.get_memory_info`.3863 * Added `tf.config.experimental.get_memory_info`, returning a dict with