MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / get_memory_growth

Method get_memory_growth

tensorflow/python/eager/context.py:1194–1201  ·  view source on GitHub ↗

Get if memory growth is enabled for a PhysicalDevice.

(self, dev)

Source from the content-addressed store, hash-verified

1192 self._visible_device_list = visible_device_list
1193
1194 def get_memory_growth(self, dev):
1195 """Get if memory growth is enabled for a PhysicalDevice."""
1196 self._initialize_physical_devices()
1197
1198 if dev not in self._physical_devices:
1199 raise ValueError("Unrecognized device: %s" % repr(dev))
1200
1201 return self._memory_growth_map[dev]
1202
1203 def set_memory_growth(self, dev, enable):
1204 """Set if memory growth should be enabled for a PhysicalDevice."""

Callers 2

get_memory_growthFunction · 0.80
testGpuGrowthMethod · 0.80

Calls 1

Tested by 1

testGpuGrowthMethod · 0.64