• L
    Add functional autograd API:hessian (#36108) · 1f93582c
    levi131 提交于
    * init functional jacobian api
    
    * finish test with dtype float32
    
    * add float64 test case
    
    * polish code
    
    * use atol=1e-5 with dtype float64
    
    * fix for ci
    
    * set timeout for test_jacobian
    
    * init hessian API
    
    * save status
    
    * polish API docstring
    
    * modify docstring
    
    * add utils.py
    
    * save status
    
    * fix dygraph double grad dtype error when calling for high differential senario
    
    * reinvoke ci
    
    * test_hessian.py is ok
    
    * polish hessian API
    
    * init vhp
    
    * Revert "init vhp"
    
    This reverts commit cbd4d3b66abe82b0ac10721b9eddeb7d82e0a1c8.
    
    * add test for partial_engine.cc
    
    * modify numerical_delta with dtype float32
    
    * merge fix for dtype float64
    
    * spell fix
    
    * polish code
    
    * rm _stop_gradient_pre_process
    Co-authored-by: NJiabinYang <360788950@qq.com>
    1f93582c
test_jacobian.py 6.1 KB