Function AddN

tensorflow/core/kernels/mkl_aggregate_ops_test.cc:38–63

template <typename T>
static Graph* AddN(const string& kind, int num_inputs, const TensorShape& shape) {
  Graph* g = new Graph(OpRegistry::Global());
  DataType type = DataTypeToEnum<T>::v();

  const bool isDefault = (kind == "Default");
  string op_name = isDefault ? "AddN" : "_MklAddN";
  std::vector<NodeBuilder::NodeOut> inputs;
  std::vector<NodeBuilder::NodeOut> inputs_not_mkl;
  inputs.reserve(num_inputs);
  inputs_not_mkl.reserve(num_inputs);

  for (int i = 0; i < num_inputs; ++i) {
    Tensor in(type, shape);
    in.flat<T>().setRandom();
    inputs.push_back(test::graph::Constant(g, in));
    inputs_not_mkl.push_back(test::graph::Constant(g, GetMklMetaTensor(), "not_mkl"));
  }

  auto nodeBuilder = NodeBuilder(g->NewName("n"), op_name)
                    .Input(inputs)
                    .Attr("T", type);
  isDefault ? nodeBuilder : nodeBuilder.Input(inputs_not_mkl)
                                       .Attr("_kernel", "MklLayoutDependentOp");
  TF_CHECK_OK(nodeBuilder.Finalize(g, nullptr));
  return g;
}

#define BM_AddN_Base(kind, T, name, nums, tf_shape, DEVICE, NTH)                          \
  static void BM_AddN##_##kind##_##T##_##name##_##DEVICE##_##NTH(                         \