We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c8f91ed commit ffd0cb0Copy full SHA for ffd0cb0
1 file changed
src/base/median_dim_values.h
@@ -25,14 +25,23 @@ class MedianDimValues : public Operator<MedianDimValues> {
25
26
protected:
27
Tensor::Shape self_shape_;
28
+
29
Tensor::Strides self_strides_;
30
31
DataType self_type_;
32
33
Tensor::Shape values_shape_;
34
35
Tensor::Strides values_strides_;
36
37
DataType values_type_;
38
39
Tensor::Shape indices_shape_;
40
41
Tensor::Strides indices_strides_;
42
43
DataType indices_type_;
44
45
int device_index_{0};
46
};
47
0 commit comments