Clean up implementation and explicitly name vars

Author: gs-olive

core/partitioning/shape_analysis.cpp

@@ -12,6 +12,11 @@ namespace partitioning {
 at::Tensor generateSingleInput(ir::Input& input, c10::optional<at::ScalarType>& type_opt) {
   auto cur_shape = input.input_shape;
   std::vector<int64_t> shape;
+
+  // Initialize min and max ranges for random number selection
+  int LoValIncl = 0;
+  int HiValExcl = 2;
+
   shape.insert(shape.begin(), std::begin(cur_shape.d), std::begin(cur_shape.d) + cur_shape.nbDims);
 
   auto type = at::kFloat;
@@ -20,7 +25,8 @@ at::Tensor generateSingleInput(ir::Input& input, c10::optional<at::ScalarType>&
   } else {
     LOG_WARNING("Input type for doing shape analysis could not be determined, defaulting to F32");
   }
-  auto in = at::randint(2, shape, {at::kCUDA}).to(type);
+
+  auto in = at::randint(LoValIncl, HiValExcl, shape, {at::kCUDA}).to(type);
 
   return in;
 }