Make default partitioning tensor input to uniform float range

- Perform casting to desired datatype after selecting a float from the
range [0, 2)

Author: gs-olive

core/partitioning/shape_analysis.cpp

@@ -26,7 +26,9 @@ at::Tensor generateSingleInput(ir::Input& input, c10::optional<at::ScalarType>&
     LOG_WARNING("Input type for doing shape analysis could not be determined, defaulting to F32");
   }
 
-  auto in = at::randint(LoValIncl, HiValExcl, shape, {at::kCUDA}).to(type);
+  // Make the value range for input tensor a uniform (float) distribution
+  // over [LoValIncl, HiValExcl), then cast to the desired dtype
+  auto in = ((LoValIncl - HiValExcl) * at::rand(shape, {at::kCUDA}) + HiValExcl).to(type);
 
   return in;
 }