Fix IndicesEstimator constructor

Signed-off-by: Yuanyuan Chen <cyyever@outlook.com>

Author: cyyever

fbgemm_gpu/src/tbe/eeg/indices_estimator.cpp

@@ -45,7 +45,7 @@ void IndicesEstimator::populateLogTable_() {
   }
 }
 
-IndicesEstimator::IndicesEstimator(const torch::Tensor& indices) {
+IndicesEstimator::init(const torch::Tensor& indices) {
   TORCH_CHECK(
       indices.numel() > 0, "indices numel is ", indices.numel(), "(< 1)");
 
@@ -62,6 +62,10 @@ IndicesEstimator::IndicesEstimator(const torch::Tensor& indices) {
   populateLogTable_();
 }
 
+IndicesEstimator::IndicesEstimator(const torch::Tensor& indices) {
+  init(indices);
+}
+
 IndicesEstimator::IndicesEstimator(const std::filesystem::path& tensors_path) {
   // NOTE: PyTorch API requires us to use a torch::pickle_load on a
   // vector<char> (torch::load doesn't work here)
@@ -79,7 +83,7 @@ IndicesEstimator::IndicesEstimator(const std::filesystem::path& tensors_path) {
   assert((ival.isTensor()) && "Loaded file is not a tensor!");
 
   // Pass it to the tensor-based constructor
-  IndicesEstimator(ival.toTensor());
+  init(ival.toTensor());
 }
 
 std::vector<double> IndicesEstimator::heavyHitters() const {

fbgemm_gpu/src/tbe/eeg/indices_estimator.h

@@ -37,6 +37,8 @@ class IndicesEstimator {
   double estimateQuality(const IndicesDistributionParameters& params) const;
 
  private:
+  void init(const torch::Tensor& indices);
+
   // Hardcoded parameters
   // Heavy hitter threshold: we make sure they account for >=90% of the mass or
   // the top 20 frequencies, whichever comes first. We also ensure that we cover