提交 52f2bc1f 编写于 作者: Q qijun

fix vector size error

上级 857cd24a
...@@ -46,7 +46,8 @@ class GaussianRandomOp : public framework::OperatorWithKernel { ...@@ -46,7 +46,8 @@ class GaussianRandomOp : public framework::OperatorWithKernel {
void InferShape(const framework::InferShapeContext& context) const override { void InferShape(const framework::InferShapeContext& context) const override {
auto* tensor = context.Output<framework::Tensor>("Out"); auto* tensor = context.Output<framework::Tensor>("Out");
auto dims = GetAttr<std::vector<int>>("dims"); auto dims = GetAttr<std::vector<int>>("dims");
std::vector<int64_t> temp(dims.size()); std::vector<int64_t> temp;
temp.reserve(dims.size());
for (auto dim : dims) { for (auto dim : dims) {
temp.push_back(static_cast<int64_t>(dim)); temp.push_back(static_cast<int64_t>(dim));
} }
......
...@@ -52,7 +52,8 @@ class UniformRandomOp : public framework::OperatorWithKernel { ...@@ -52,7 +52,8 @@ class UniformRandomOp : public framework::OperatorWithKernel {
"uniform_random's min must less then max"); "uniform_random's min must less then max");
auto* tensor = ctx.Output<framework::Tensor>("Out"); auto* tensor = ctx.Output<framework::Tensor>("Out");
auto dims = GetAttr<std::vector<int>>("dims"); auto dims = GetAttr<std::vector<int>>("dims");
std::vector<int64_t> temp(dims.size()); std::vector<int64_t> temp;
temp.reserve(dims.size());
for (auto dim : dims) { for (auto dim : dims) {
temp.push_back(static_cast<int64_t>(dim)); temp.push_back(static_cast<int64_t>(dim));
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册