Searched refs:AvailableGPUMemory (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ | ||
| D | devices.h | 34 int64 AvailableGPUMemory(int gpu_id); |
| D | devices.cc | 60 int64 AvailableGPUMemory(int gpu_id) { in AvailableGPUMemory() function |