Searched refs:GetScopedAllocatorMgr (Results 1 – 8 of 8) sorted by relevance
66 ScopedAllocatorMgr* GetScopedAllocatorMgr() const override { in GetScopedAllocatorMgr() function67 return underlying_device_->GetScopedAllocatorMgr(); in GetScopedAllocatorMgr()
35 ScopedAllocatorMgr* GetScopedAllocatorMgr() const override { in GetScopedAllocatorMgr() function
1773 ScopedAllocatorMgr* sam = d->GetScopedAllocatorMgr(); in RunState()
174 device_->GetScopedAllocatorMgr(), "concat", &tensors, &inputs_, in ExecOp()293 device_->GetScopedAllocatorMgr(), "split", &tensors, &inputs_, in ExecOp()
51 ScopedAllocatorMgr* sam = device_->GetScopedAllocatorMgr(); in Compute()
201 virtual ScopedAllocatorMgr* GetScopedAllocatorMgr() const { return nullptr; } in GetScopedAllocatorMgr() function
102 ScopedAllocatorMgr* GetScopedAllocatorMgr() const override { in GetScopedAllocatorMgr() function
360 ScopedAllocatorMgr* sam = d->GetScopedAllocatorMgr(); in CleanupGraphAsync()