We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 79c91bf commit a3a4c73Copy full SHA for a3a4c73
1 file changed
src/base/nll_loss2d_backward_grad_input.h
@@ -35,20 +35,35 @@ class NllLoss2dBackwardGradInput : public Operator<NllLoss2dBackwardGradInput> {
35
36
protected:
37
Tensor::Shape grad_output_shape_;
38
+
39
Tensor::Strides grad_output_strides_;
40
41
DataType grad_output_type_;
42
43
Tensor::Shape self_shape_;
44
45
Tensor::Strides self_strides_;
46
47
DataType self_type_;
48
49
Tensor::Shape target_shape_;
50
51
Tensor::Strides target_strides_;
52
53
DataType target_type_;
54
55
Tensor::Shape total_weight_shape_;
56
57
Tensor::Strides total_weight_strides_;
58
59
DataType total_weight_type_;
60
61
Tensor::Shape grad_input_shape_;
62
63
Tensor::Strides grad_input_strides_;
64
65
DataType grad_input_type_;
66
67
int device_index_{0};
68
};
69
0 commit comments