Searched defs:memory_fraction (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ | ||
| D | gpu_device.h | 51 double memory_fraction = 0.9; member |
| D | gpu_device.cc | 232 double memory_fraction, bool preallocate) { in CreateBFCAllocator() |