File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -196,9 +196,9 @@ template <typename DeviceContext, typename T>
196196class BeamSearchOpKernel : public framework ::OpKernel<T> {
197197 public:
198198 void Compute (const framework::ExecutionContext& context) const override {
199- auto ids_var = context.Input <framework::LoDTensor>(" ids" );
200- auto scores_var = context.Input <framework::LoDTensor>(" scores" );
201- auto pre_ids_var = context.Input <framework::LoDTensor>(" pre_ids" );
199+ auto * ids_var = context.Input <framework::LoDTensor>(" ids" );
200+ auto * scores_var = context.Input <framework::LoDTensor>(" scores" );
201+ auto * pre_ids_var = context.Input <framework::LoDTensor>(" pre_ids" );
202202 PADDLE_ENFORCE_NOT_NULL (ids_var);
203203 PADDLE_ENFORCE_NOT_NULL (scores_var);
204204 PADDLE_ENFORCE_NOT_NULL (pre_ids_var);
You can’t perform that action at this time.
0 commit comments