File tree 1 file changed +9
-9
lines changed
python/paddle/utils/code_gen
1 file changed +9
-9
lines changed Original file line number Diff line number Diff line change 633
633
output : Tensor(x_grad)
634
634
infer_meta :
635
635
func : UnchangedInferMeta
636
- param : [out ]
636
+ param : [x ]
637
637
kernel :
638
638
func : psroi_pool_grad
639
639
optional : rois_num
650
650
func : put_along_axis_grad
651
651
652
652
- backward_api : relu_double_grad
653
- forward : relu_grad (tensor out, tensor grad_out) -> tensor (grad_x)
654
- args : (tensor out, tensor grad_x_grad)
655
- output : tensor (out_grad), tensor (grad_out_grad)
653
+ forward : relu_grad (Tensor out, Tensor grad_out) -> Tensor (grad_x)
654
+ args : (Tensor out, Tensor grad_x_grad)
655
+ output : Tensor (out_grad), Tensor (grad_out_grad)
656
656
infer_meta :
657
- func : generalbinarygradinfermeta
657
+ func : GeneralBinaryGradInferMeta
658
658
param : [out, out]
659
659
kernel :
660
660
func : relu_double_grad
661
661
662
662
- backward_api : relu_grad
663
- forward : relu (tensor x) -> tensor (out)
664
- args : (tensor out, tensor out_grad)
665
- output : tensor (x_grad)
663
+ forward : relu (Tensor x) -> Tensor (out)
664
+ args : (Tensor out, Tensor out_grad)
665
+ output : Tensor (x_grad)
666
666
infer_meta :
667
- func : unchangedinfermeta
667
+ func : UnchangedInferMeta
668
668
param : [out]
669
669
kernel :
670
670
func : relu_grad
You can’t perform that action at this time.
0 commit comments