Searched defs:MemoryReclaimContext (Results 1 – 1 of 1) sorted by relevance
130 struct MemoryReclaimContext { struct131 MemoryReclaimContext(xla::Backend* backend, int device_ordinal, in MemoryReclaimContext() argument151 Status TryFreeMemoryStep(MemoryReclaimContext* mrctx, const Status& status); argument