Small changes to optimizer and utils.
Adds a `var_list` arg to optimizer. Removes call to `convert_to_tensor` from _momentum to support non-default dtypes. Implements `posdef_inv_eig` for matrix inversion. PiperOrigin-RevId: 176690230
Loading
Please sign in to comment