We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent fbb35e7 commit eca4abdCopy full SHA for eca4abd
1 file changed
src/base/gelu_backward_grad_input.h
@@ -25,14 +25,23 @@ class GeluBackwardGradInput : public Operator<GeluBackwardGradInput> {
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