We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3697dc6 commit e6d69c7Copy full SHA for e6d69c7
1 file changed
src/base/fractional_max_pool2d_backward_grad_input.h
@@ -34,17 +34,29 @@ class FractionalMaxPool2dBackwardGradInput
34
35
protected:
36
Tensor::Shape grad_output_shape_;
37
+
38
Tensor::Strides grad_output_strides_;
39
40
DataType grad_output_type_;
41
42
Tensor::Shape self_shape_;
43
44
Tensor::Strides self_strides_;
45
46
DataType self_type_;
47
48
Tensor::Shape indices_shape_;
49
50
Tensor::Strides indices_strides_;
51
52
DataType indices_type_;
53
54
Tensor::Shape grad_input_shape_;
55
56
Tensor::Strides grad_input_strides_;
57
58
DataType grad_input_type_;
59
60
int device_index_{0};
61
};
62
0 commit comments