Searched refs:AllocationTracker (Results 1 – 3 of 3) sorted by relevance
34 StatusOr<GlobalDataHandle> AllocationTracker::Register( in Register()43 StatusOr<GlobalDataHandle> AllocationTracker::RegisterReplicatedBuffers( in RegisterReplicatedBuffers()59 StatusOr<GlobalDataHandle> AllocationTracker::RegisterInternal( in RegisterInternal()96 Status AllocationTracker::Unregister(const GlobalDataHandle& data) { in Unregister()128 StatusOr<std::vector<GlobalDataHandle>> AllocationTracker::DeconstructTuple( in DeconstructTuple()166 StatusOr<std::vector<const ShapedBuffer*>> AllocationTracker::Resolve( in Resolve()169 return AllocationTracker::ResolveInternal(data); in Resolve()172 StatusOr<const ShapedBuffer*> AllocationTracker::ResolveForReplica( in ResolveForReplica()186 StatusOr<std::vector<const ShapedBuffer*>> AllocationTracker::ResolveInternal( in ResolveInternal()206 void AllocationTracker::AddAllocationOrIncrementRefCount( in AddAllocationOrIncrementRefCount()[all …]
39 class AllocationTracker {44 AllocationTracker(Backend* backend) : backend_(backend), next_handle_(1) {} in AllocationTracker() function151 TF_DISALLOW_COPY_AND_ASSIGN(AllocationTracker);
293 AllocationTracker allocation_tracker_;