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>
196
196
class BeamSearchOpKernel : public framework ::OpKernel<T> {
197
197
public:
198
198
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" );
202
202
PADDLE_ENFORCE_NOT_NULL (ids_var);
203
203
PADDLE_ENFORCE_NOT_NULL (scores_var);
204
204
PADDLE_ENFORCE_NOT_NULL (pre_ids_var);
You can’t perform that action at this time.
0 commit comments