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