We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a102efa commit bdc4398Copy full SHA for bdc4398
1 file changed
src/base/fractional_max_pool3d_output.h
@@ -34,17 +34,29 @@ class FractionalMaxPool3dOutput : public Operator<FractionalMaxPool3dOutput> {
34
35
protected:
36
Tensor::Shape self_shape_;
37
+
38
Tensor::Strides self_strides_;
39
40
DataType self_type_;
41
42
Tensor::Shape random_samples_shape_;
43
44
Tensor::Strides random_samples_strides_;
45
46
DataType random_samples_type_;
47
48
Tensor::Shape output_shape_;
49
50
Tensor::Strides output_strides_;
51
52
DataType output_type_;
53
54
Tensor::Shape indices_shape_;
55
56
Tensor::Strides indices_strides_;
57
58
DataType indices_type_;
59
60
int device_index_{0};
61
};
62
0 commit comments