Lml/vhp (#36146)
* 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.
* init vhp
* finish vhp API logically
* add test for partial_engine.cc
* modify numerical_delta with dtype float32
* merge fix for dtype float64
* spell fix
* save status
* polish code
* rm _stop_gradient_pre_process
* save status
* add example for vhp interface
* add _compute_numerical_vjp and _compute_numerical_vhp
* test is ok
* vhp is ok
* add testVHPFloat64
* modify for comments
* modify format
* modify format
* save status
* test_vhp is ok
* finish code polish
* small modify for v is None
Co-authored-by: NJiabinYang <360788950@qq.com>
Showing
想要评论请 注册 或 登录