提交 f67897a2 编写于 作者: J jingqinghe

fix bug

上级 b2a724a8
...@@ -106,9 +106,9 @@ template <typename DeviceContext, typename T> ...@@ -106,9 +106,9 @@ template <typename DeviceContext, typename T>
class MpcElementwiseAddKernel : public MpcOpKernel<T> { class MpcElementwiseAddKernel : public MpcOpKernel<T> {
public: public:
void ComputeImpl(const framework::ExecutionContext &ctx) const override { void ComputeImpl(const framework::ExecutionContext &ctx) const override {
auto *in_x_t = ctx.Input<LoDTensor>("X"); auto *in_x_t = ctx.Input<Tensor>("X");
auto *in_y_t = ctx.Input<LoDTensor>("Y"); auto *in_y_t = ctx.Input<Tensor>("Y");
auto *out_t = ctx.Output<LoDTensor>("Out"); auto *out_t = ctx.Output<Tensor>("Out");
int axis = ctx.Attr<int>("axis"); int axis = ctx.Attr<int>("axis");
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册