Searched refs:set_memory_growth (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/framework/ |
| D | config.py | 637 def set_memory_growth(device, enable): function
661 context.context().set_memory_growth(device, enable)
|
| D | config_test.py | 579 config.set_memory_growth(gpu, True)
590 config.set_memory_growth(gpu, False)
594 config.set_memory_growth(gpu, True)
656 config.set_memory_growth(gpus[0], True)
676 config.set_memory_growth(gpu, True)
697 config.set_memory_growth(gpus[-1], False)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.config.experimental.pbtxt | 76 name: "set_memory_growth"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.config.experimental.pbtxt | 76 name: "set_memory_growth"
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 1461 def set_memory_growth(self, dev, enable): member in Context
|