Skip to content

Commit d50a8b9

Browse files
committed
chore: separate base class members
1 parent 0ef689a commit d50a8b9

1 file changed

Lines changed: 12 additions & 0 deletions

File tree

src/base/nll_loss2d_forward_output.h

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,17 +30,29 @@ class NllLoss2dForwardOutput : public Operator<NllLoss2dForwardOutput> {
3030

3131
protected:
3232
Tensor::Shape self_shape_;
33+
3334
Tensor::Strides self_strides_;
35+
3436
DataType self_type_;
37+
3538
Tensor::Shape target_shape_;
39+
3640
Tensor::Strides target_strides_;
41+
3742
DataType target_type_;
43+
3844
Tensor::Shape output_shape_;
45+
3946
Tensor::Strides output_strides_;
47+
4048
DataType output_type_;
49+
4150
Tensor::Shape total_weight_shape_;
51+
4252
Tensor::Strides total_weight_strides_;
53+
4354
DataType total_weight_type_;
55+
4456
int device_index_{0};
4557
};
4658

0 commit comments

Comments
 (0)