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