Searched refs:allocation_tracker_ (Results 1 – 3 of 3) sorted by relevance
161 allocation_tracker_(execute_backend.get()), in Service()197 Status unregister_status = allocation_tracker_.Unregister(data); in Unregister()210 allocation_tracker_.DeconstructTuple(arg->tuple_handle())); in DeconstructTuple()239 auto buffer_status = allocation_tracker_.Resolve(*arguments[i]); in ResolveAndValidateArguments()425 allocation_tracker_.RegisterReplicatedBuffers( in ExecuteParallelAndRegisterResult()508 return allocation_tracker_.Register(std::move(result), result_tag); in ExecuteAndRegisterResult()521 return allocation_tracker_.RegisterReplicatedBuffers(std::move(results), in ExecuteAndRegisterResult()718 allocation_tracker_.ResolveForReplica(outputs[i], 0)); in ExecuteGraphParallel()890 allocation_tracker_.ResolveForReplica(result->output(), 0)); in Execute()919 allocation_tracker_.ResolveForReplica(arg->data(), 0)); in TransferToClient()[all …]
198 TF_ASSIGN_OR_RETURN(auto buffers, allocation_tracker_.Resolve(data)); in GlobalDataToShapedBuffer()209 return allocation_tracker_.RegisterReplicatedBuffers( in RegisterReplicatedBuffers()
293 AllocationTracker allocation_tracker_; variable