We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2edd342 commit 255e5feCopy full SHA for 255e5fe
1 file changed
src/base/multilabel_margin_loss_backward_grad_input.h
@@ -36,20 +36,35 @@ class MultilabelMarginLossBackwardGradInput
36
37
protected:
38
Tensor::Shape grad_output_shape_;
39
+
40
Tensor::Strides grad_output_strides_;
41
42
DataType grad_output_type_;
43
44
Tensor::Shape self_shape_;
45
46
Tensor::Strides self_strides_;
47
48
DataType self_type_;
49
50
Tensor::Shape target_shape_;
51
52
Tensor::Strides target_strides_;
53
54
DataType target_type_;
55
56
Tensor::Shape is_target_shape_;
57
58
Tensor::Strides is_target_strides_;
59
60
DataType is_target_type_;
61
62
Tensor::Shape grad_input_shape_;
63
64
Tensor::Strides grad_input_strides_;
65
66
DataType grad_input_type_;
67
68
int device_index_{0};
69
};
70
0 commit comments