Function CubeWithMulSquare

tensorflow/core/kernels/cwise_ops_test.cc:172–180

}

Graph* CubeWithMulSquare(int num) {
  Graph* g = new Graph(OpRegistry::Global());
  Tensor lhs(DT_FLOAT, TensorShape({64, 64, num / (64 * 64)}));
  lhs.flat<float>().setRandom();
  auto* x = test::graph::Constant(g, lhs);
  auto* inner = test::graph::Unary(g, "Square", x);
  test::graph::Binary(g, "Mul", test::graph::Constant(g, lhs), inner);
  return g;
}

#define BM_CUBE(DEVICE, Impl)                          \
  void BM_##DEVICE##_Cube_##Impl(int iters, int num) { \