Function BM_Construct

tensorflow/compiler/xla/shape_tree_test.cc:514–526

}

void BM_Construct(int iters, int depth, int fan_out) {
  tensorflow::testing::StopTiming();
  Shape shape = ShapeUtil::MakeShape(F32, {32, 64, 128});
  for (int i = 0; i < depth; ++i) {
    std::vector<xla::Shape> shapes(fan_out, shape);
    shape = ShapeUtil::MakeTupleShape(shapes);
  }
  tensorflow::testing::StartTiming();

  for (int i = 0; i < iters; ++i) {
    ShapeTree<int> shape_tree(shape);
  }
}

void BM_ConstructUnowned(int iters, int depth, int fan_out) {
  tensorflow::testing::StopTiming();