Function BM_RegisterOneByOne

tensorflow/core/util/tensor_slice_set_test.cc:147–154

}

static void BM_RegisterOneByOne(int parts) {
  TensorShape shape({parts, 41});
  TensorSliceSet slice_set(shape, DT_INT32);
  for (int i = 0; i < parts; ++i) {
    TensorSlice part({{i, 1}, {0, -1}});
    TF_CHECK_OK(slice_set.Register(part, part.DebugString()));
  }
}

BENCHMARK(BM_RegisterOneByOne);