We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 43933ef commit cdcf25dCopy full SHA for cdcf25d
1 file changed
src/base/nanmedian_dim_values.h
@@ -25,14 +25,23 @@ class NanmedianDimValues : public Operator<NanmedianDimValues> {
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