Searched refs:XRTMemoryManager (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_memory_manager.cc | 48 class XRTMemoryManager::DeviceContext {
101 Status CompactAllocations(XRTMemoryManager* memory_manager, in CompactAllocations()
187 XRTMemoryManager::WorkingSet::WorkingSet( in WorkingSet()
188 RefPtr<XRTMemoryManager> memory_manager) in WorkingSet()
191 XRTMemoryManager::WorkingSet::~WorkingSet() { in ~WorkingSet()
197 Status XRTMemoryManager::WorkingSet::LookupAndPin(xla::Backend* backend, in LookupAndPin()
206 /* static */ RefPtr<XRTMemoryManager> XRTMemoryManager::Get(ResourceMgr* rm) { in Get()
209 XRTMemoryManager* memory_manager = nullptr; in Get()
210 TF_CHECK_OK(rm->LookupOrCreate<XRTMemoryManager>( in Get()
211 *container, *name, &memory_manager, [](XRTMemoryManager** ret) { in Get()
[all …]
|
| D | xrt_memory_manager.h | 41 class XRTMemoryManager : public ResourceBase {
53 explicit WorkingSet(RefPtr<XRTMemoryManager> memory_manager);
65 const RefPtr<XRTMemoryManager>& MemoryManager() const { in MemoryManager()
70 RefPtr<XRTMemoryManager> memory_manager_;
75 static RefPtr<XRTMemoryManager> Get(ResourceMgr* rm);
158 xla::StatusOr<T> XRTMemoryManager::Run( in Run()
|
| D | xrt_state.h | 45 class XRTMemoryManager; variable
84 XRTMemoryManager* memory_manager,
92 XRTMemoryManager* memory_manager,
139 static Status MakeTuple(XRTMemoryManager* memory_manager,
160 xla::StatusOr<bool> SwapIn(XRTMemoryManager* memory_manager,
167 xla::StatusOr<bool> PinAndSwapIn(XRTMemoryManager* memory_manager,
|
| D | xrt_util.cc | 40 explicit ScopedHandles(RefPtr<XRTMemoryManager> memory_manager) in ScopedHandles()
60 handles_.resize(index + 1, XRTMemoryManager::InvalidKey()); in Add()
97 handles_[index] = XRTMemoryManager::InvalidKey(); in Release()
109 RefPtr<XRTMemoryManager> memory_manager_;
153 XRTMemoryManager::WorkingSet* working_set) { in PopulateOpWorkingSet()
257 XRTMemoryManager::WorkingSet* working_set, xla::Backend* backend, in GetInputTupleAllocations()
346 XRTMemoryManager* memory_manager, in CreateExecuteOutput()
374 const RefPtr<XRTMemoryManager>& memory_manager, in ExecuteChained()
401 XRTMemoryManager::WorkingSet working_set(memory_manager); in ExecuteChained()
|
| D | xrt_util.h | 79 XRTMemoryManager::WorkingSet* working_set, xla::Backend* backend,
97 XRTMemoryManager* memory_manager,
108 const RefPtr<XRTMemoryManager>& memory_manager,
|
| D | xrt_state.cc | 73 XRTMemoryManager* memory_manager, xla::Backend* backend, int device_ordinal, in AllocateScopedShapedBuffer()
170 const xla::LiteralBase& literal, XRTMemoryManager* memory_manager, in CreateAndTransfer()
198 const xla::Shape& shape, XRTMemoryManager* memory_manager, in CreateUninitialized()
292 xla::StatusOr<bool> XRTTupleAllocation::SwapIn(XRTMemoryManager* memory_manager, in SwapIn()
327 XRTMemoryManager* memory_manager, xla::Backend* backend) { in PinAndSwapIn()
455 XRTMemoryManager* memory_manager, xla::Backend* backend, int device_ordinal, in MakeTuple()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 111 XRTMemoryManager::Get(rm)->Lookup(key)); in ParseTupleNode()
200 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute()
240 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute()
344 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute()
393 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute()
463 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute()
506 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute()
559 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute()
652 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute()
686 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in Compute()
[all …]
|
| D | xrt_execute_op.cc | 89 xla::LocalExecutable* executable, XRTMemoryManager::WorkingSet* working_set, in GetInputTuples()
375 OpKernelContext* context, XRTMemoryManager* memory_manager, in ExecuteComputation()
395 OpKernelContext* context, const RefPtr<XRTMemoryManager>& memory_manager, in ExecuteComputation()
401 XRTMemoryManager::WorkingSet working_set(memory_manager); in ExecuteComputation()
474 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in DoWork()
557 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in DoWork()
|
| D | tpu_execute_op.cc | 171 OpKernelContext* context, XRTMemoryManager* memory_manager, in AllocateOutputTensors()
231 XRTMemoryManager* memory_manager, const TPUExecutableInfoProto& executable, in ExecuteTPUProgram()
340 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in DoWork()
341 XRTMemoryManager::WorkingSet working_set(memory_manager); in DoWork()
427 RefPtr<XRTMemoryManager> memory_manager = XRTMemoryManager::Get(rm); in DoWork()
|