We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 657656a commit 104425aCopy full SHA for 104425a
1 file changed
src/base/reflection_pad1d_backward_grad_input.h
@@ -28,14 +28,23 @@ class ReflectionPad1dBackwardGradInput
28
29
protected:
30
Tensor::Shape grad_output_shape_;
31
+
32
Tensor::Strides grad_output_strides_;
33
34
DataType grad_output_type_;
35
36
Tensor::Shape self_shape_;
37
38
Tensor::Strides self_strides_;
39
40
DataType self_type_;
41
42
Tensor::Shape grad_input_shape_;
43
44
Tensor::Strides grad_input_strides_;
45
46
DataType grad_input_type_;
47
48
int device_index_{0};
49
};
50
0 commit comments