We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 5ebc0a1 commit ed0dbfaCopy full SHA for ed0dbfa
1 file changed
src/base/nll_loss_forward_output.h
@@ -30,17 +30,29 @@ class NllLossForwardOutput : public Operator<NllLossForwardOutput> {
30
31
protected:
32
Tensor::Shape self_shape_;
33
+
34
Tensor::Strides self_strides_;
35
36
DataType self_type_;
37
38
Tensor::Shape target_shape_;
39
40
Tensor::Strides target_strides_;
41
42
DataType target_type_;
43
44
Tensor::Shape output_shape_;
45
46
Tensor::Strides output_strides_;
47
48
DataType output_type_;
49
50
Tensor::Shape total_weight_shape_;
51
52
Tensor::Strides total_weight_strides_;
53
54
DataType total_weight_type_;
55
56
int device_index_{0};
57
};
58
0 commit comments