We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8f6ad61 commit 3bec5f0Copy full SHA for 3bec5f0
1 file changed
src/base/nll_loss_backward_grad_input.h
@@ -35,20 +35,35 @@ class NllLossBackwardGradInput : public Operator<NllLossBackwardGradInput> {
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