We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c989c3b commit edc2777Copy full SHA for edc2777
1 file changed
src/base/multilabel_margin_loss_forward_output.h
@@ -31,17 +31,29 @@ class MultilabelMarginLossForwardOutput
31
32
protected:
33
Tensor::Shape self_shape_;
34
+
35
Tensor::Strides self_strides_;
36
37
DataType self_type_;
38
39
Tensor::Shape target_shape_;
40
41
Tensor::Strides target_strides_;
42
43
DataType target_type_;
44
45
Tensor::Shape output_shape_;
46
47
Tensor::Strides output_strides_;
48
49
DataType output_type_;
50
51
Tensor::Shape is_target_shape_;
52
53
Tensor::Strides is_target_strides_;
54
55
DataType is_target_type_;
56
57
int device_index_{0};
58
};
59
0 commit comments