Searched refs:RegisterReplicatedBuffers (Results 1 – 6 of 6) sorted by relevance
200 StatusOr<GlobalDataHandle> LocalService::RegisterReplicatedBuffers( in RegisterReplicatedBuffers() function in xla::LocalService202 return allocation_tracker_.RegisterReplicatedBuffers( in RegisterReplicatedBuffers()
68 StatusOr<GlobalDataHandle> RegisterReplicatedBuffers(
54 StatusOr<GlobalDataHandle> RegisterReplicatedBuffers(
42 StatusOr<GlobalDataHandle> AllocationTracker::RegisterReplicatedBuffers( in RegisterReplicatedBuffers() function in xla::AllocationTracker
469 allocation_tracker_.RegisterReplicatedBuffers( in ExecuteParallelAndRegisterResult()569 return allocation_tracker_.RegisterReplicatedBuffers(std::move(results), in ExecuteAndRegisterResult()1001 allocation_tracker_.RegisterReplicatedBuffers( in TransferToServer()
329 local_service_->RegisterReplicatedBuffers( in TransferToLocalServer()