Home
last modified time | relevance | path

Searched refs:MemoryReclaimContext (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_memory_manager.h130 struct MemoryReclaimContext { struct
131 MemoryReclaimContext(xla::Backend* backend, int device_ordinal, in MemoryReclaimContext() function
151 Status TryFreeMemoryStep(MemoryReclaimContext* mrctx, const Status& status); argument
161 MemoryReclaimContext mrctx(backend, device_ordinal, requested_free_size); in Run()
Dxrt_memory_manager.cc321 Status XRTMemoryManager::TryFreeMemoryStep(MemoryReclaimContext* mrctx, in TryFreeMemoryStep()