Searched refs:GetScopedAllocatorMgr (Results 1 – 8 of 8) sorted by relevance
67 ScopedAllocatorMgr* GetScopedAllocatorMgr() const override { in GetScopedAllocatorMgr() function68 return underlying_->GetScopedAllocatorMgr(); in GetScopedAllocatorMgr()
35 ScopedAllocatorMgr* GetScopedAllocatorMgr() const override { in GetScopedAllocatorMgr() function
1650 ScopedAllocatorMgr* sam = d->GetScopedAllocatorMgr(); in RunState()
173 device_->GetScopedAllocatorMgr(), "concat", &tensors, &inputs_, in ExecOp()292 device_->GetScopedAllocatorMgr(), "split", &tensors, &inputs_, in ExecOp()
51 ScopedAllocatorMgr* sam = device_->GetScopedAllocatorMgr(); in Compute()
202 virtual ScopedAllocatorMgr* GetScopedAllocatorMgr() const { return nullptr; } in GetScopedAllocatorMgr() function
115 ScopedAllocatorMgr* GetScopedAllocatorMgr() const override { in GetScopedAllocatorMgr() function
355 ScopedAllocatorMgr* sam = d->GetScopedAllocatorMgr(); in CleanupGraphAsync()