Function RoundFloatToBFloat16

tensorflow/core/framework/bfloat16_test.cc:212–216

BENCHMARK(BM_FloatToBFloat16);

void RoundFloatToBFloat16(const float* src, bfloat16* dst, int64 size) {
  for (; size != 0; size--) {
    dst[size] = bfloat16(src[size]);
  }
}

static void BM_RoundFloatToBFloat16(int iters) {
  testing::StopTiming();