Function VecAlphas

tensorflow/core/kernels/random_op_test.cc:76–85

BM_RNG(gpu, TruncatedNormal);

Tensor VecAlphas(int64 n) {
  Tensor alphas(DT_DOUBLE, TensorShape({n}));
  for (int i = 0; i < n; i++) {
    // Alternate back and forth between small-and-growing (.25) and
    // large-and-shrinking (26.67) alpha.
    alphas.vec<double>()(i) =
        0.25 + MathUtil::IPow(1.1, i % 2 == 0 ? i : n - i);
  }
  return alphas;
}

void BM_cpu_RandomGamma(int iters, int nsamp, int nalpha) {
  testing::ItemsProcessed(static_cast<int64>(iters) * nsamp * nalpha);