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