We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0736163 commit 85e0d4cCopy full SHA for 85e0d4c
1 file changed
src/base/glu_backward_grad_input.h
@@ -25,14 +25,23 @@ class GluBackwardGradInput : public Operator<GluBackwardGradInput> {
25
26
protected:
27
Tensor::Shape grad_output_shape_;
28
+
29
Tensor::Strides grad_output_strides_;
30
31
DataType grad_output_type_;
32
33
Tensor::Shape self_shape_;
34
35
Tensor::Strides self_strides_;
36
37
DataType self_type_;
38
39
Tensor::Shape grad_input_shape_;
40
41
Tensor::Strides grad_input_strides_;
42
43
DataType grad_input_type_;
44
45
int device_index_{0};
46
};
47
0 commit comments