message: Add back sample limit check I inadvertently removed, which in some cases led
  to writing to unallocated memory.
diff: @@ -196,6 +196,9 @@ struct TruncatedNormalFunctor<CPUDevice, T> {
  }
  output(sample) = z * stddev + mean;
  sample++;
  + if (sample >= limit_sample) {
  + break;
  + }
  numIterations = 0;
  } else {
  numIterations++;
  