We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b94e29e commit df226a9Copy full SHA for df226a9
1 file changed
src/base/multi_margin_loss_backward_grad_input.h
@@ -31,17 +31,29 @@ class MultiMarginLossBackwardGradInput
31
32
protected:
33
Tensor::Shape grad_output_shape_;
34
+
35
Tensor::Strides grad_output_strides_;
36
37
DataType grad_output_type_;
38
39
Tensor::Shape self_shape_;
40
41
Tensor::Strides self_strides_;
42
43
DataType self_type_;
44
45
Tensor::Shape target_shape_;
46
47
Tensor::Strides target_strides_;
48
49
DataType target_type_;
50
51
Tensor::Shape grad_input_shape_;
52
53
Tensor::Strides grad_input_strides_;
54
55
DataType grad_input_type_;
56
57
int device_index_{0};
58
};
59
0 commit comments