Method ReduceSum

tensorflow/core/ops/math_grad_test.cc:228–249

  }

  Tensor ReduceSum(const Tensor& x, gtl::ArraySlice<int32> axes) {
    int num_axes = axes.length();
    Tensor y(DT_INT32, TensorShape({num_axes}));
    for (size_t i = 0; i < axes.size(); ++i) {
      y.flat<int32>()(i) = axes[i];
    }
    auto T = x.dtype();
    auto gdef = test::function::GDef(
        {
            f::NDef("x", "Placeholder", {}, {{"dtype", T}}),
            f::NDef("y", "Const", {}, {{"dtype", DT_INT32}, {"value", y}}),
            f::NDef("z", "Sum", {"x", "y"}, {{"T", T}}),
        },
        {});
    auto sess = NewSession();
    TF_CHECK_OK(sess->Create(gdef));
    std::vector<Tensor> outputs;
    TF_CHECK_OK(sess->Run({{"x:0", x}}, {"z:0"}, {}, &outputs));
    CHECK_EQ(outputs.size(), 1);
    TF_CHECK_OK(sess->Close());
    return outputs[0];
  }

  Tensor MatMulCommon(const string& opname, const string& attr_adj_x,
                      const string& attr_adj_y, const Tensor& x, bool ax,