linearize.hpp File Reference

#include <graphite/ops/common.hpp>

Include dependency graph for linearize.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions
template<typename T , typename S , size_t I, size_t N, typename M , size_t E, typename F , typename VT , std::size_t... Is>
__global__ void	graphite::ops::compute_jacobian_kernel (const M *obs, T *error, S *jacs, const typename F::ConstraintDataType *constraint_data, const size_t *active_ids, const size_t *ids, const size_t num_threads, const VT args, const uint8_t *active_state, std::index_sequence< Is... >)
template<typename T , typename S , typename F , typename VT , std::size_t... Is>
void	graphite::ops::launch_kernel_jacobians (F *f, std::array< const size_t *, F::get_num_vertices()> &hessian_ids, VT &verts, std::array< S *, F::get_num_vertices()> &jacs, const size_t num_factors, StreamPool &streams, std::index_sequence< Is... >)
template<typename T , typename S , typename F >
void	graphite::ops::compute_jacobians (F *f, StreamPool &streams)
template<typename highp , typename T , size_t I, size_t N, size_t E, size_t D>
__global__ void	graphite::ops::scale_jacobians_kernel (T *jacs, const highp *jacobian_scales, const size_t *active_ids, const size_t *ids, const size_t *hessian_ids, const uint8_t *active_state, const size_t num_threads)
template<typename T , typename S , typename F , std::size_t... Is>
void	graphite::ops::launch_kernel_scale_jacobians (F *f, T *jacobian_scales, std::array< const size_t *, F::get_num_vertices()> &hessian_ids, std::array< S *, F::get_num_vertices()> &jacs, const size_t num_factors, std::index_sequence< Is... >)
template<typename T , typename S , typename F >
void	graphite::ops::scale_jacobians (F *f, T *jacobian_scales)
template<typename T , typename S , size_t I, size_t N, size_t E, typename F , std::size_t... Is>
__global__ void	graphite::ops::compute_b_kernel (T *b, const T *error, const size_t *active_ids, const size_t *ids, const size_t *hessian_ids, const size_t num_threads, const S *jacs, const uint8_t *active_state, const S *pmat, const S *loss_derivative, std::index_sequence< Is... >)
template<typename T , typename S , size_t I, size_t N, typename M , size_t E, typename F , typename VT , std::size_t... Is>
__global__ void	graphite::ops::compute_b_dynamic_kernel (T *b, const T *error, const size_t *active_ids, const size_t *ids, const size_t *hessian_ids, const size_t num_threads, const VT args, const M *obs, const T *jacobian_scales, const typename F::ConstraintDataType *constraint_data, const uint8_t *active_state, const S *pmat, const S *loss_derivative, std::index_sequence< Is... >)
template<typename T , typename S , typename F , std::size_t... Is>
void	graphite::ops::launch_kernel_compute_b (F *f, T *b, std::array< const size_t *, F::get_num_vertices()> &hessian_ids, std::array< S *, F::get_num_vertices()> &jacs, const T *jacobian_scales, const size_t num_factors, std::index_sequence< Is... >)
template<typename T , typename S , typename F >
void	graphite::ops::compute_b_async (F *f, T *b, const T *jacobian_scales)