[FEA] KMeans - int64_t Extents for Host (Streaming) Fit Function

[tarang-jain]

#1886 adds the streaming batched host fit API for kmeans. This issue is to track updating the APIs to use int64_t extents to accommodate much larger input matrices.