Function Square

tensorflow/compiler/xla/tests/prng_test.cc:141–143

namespace {
template <typename T>
T Square(T x) {
  return x * x;
}
}  // namespace

double PrngTest::UniformChiSquared(int32 range_size, int32 expected_count,