“506092146d05aa94d2430c5b3dfb63d699dca858”上不存在“paddle/phi/kernels/unsqueeze_grad_kernel.h”
提交 9c83b5a2 编写于 作者: K Kavya Srinet

Fixed uniform_random_op.cc

上级 07064355
......@@ -66,7 +66,7 @@ class UniformRandomOp : public framework::OperatorWithKernel {
framework::OpKernelType GetExpectedKernelType(
const framework::ExecutionContext& ctx) const override {
return framework::OpKernelType(
static_cast<framework::proto::DataType>(ctx.Attr<int>("dtype")),
static_cast<framework::proto::VarType::Type>(ctx.Attr<int>("dtype")),
ctx.GetPlace());
}
};
......@@ -99,7 +99,7 @@ uniform distribution.
"0 means use a seed generated by the system.")
.SetDefault(0);
AddAttr<int>("dtype", "(int, default 5(FP32)) Output tensor data type")
.SetDefault(framework::proto::DataType::FP32);
.SetDefault(framework::proto::VarType::Type::FP32);
}
};
} // namespace operators
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册