Graphite
Loading...
Searching...
No Matches
graphite::Hessian< T, S > Class Template Reference

Public Member Functions

std::vector< BlockCoordinatesget_block_coordinates (Graph< T, S > *graph)
 
void build_indices (Graph< T, S > *graph, const std::vector< BlockCoordinates > &block_coords)
 
void backup_diagonal (Graph< T, S > *graph, StreamPool &streams)
 
void apply_damping (Graph< T, S > *graph, T damping_factor, StreamPool &streams)
 
void setup_hessian_computation (Graph< T, S > *graph, StreamPool &streams)
 
void execute_hessian_computation (Graph< T, S > *graph, StreamPool &streams)
 
void build_structure (Graph< T, S > *graph, StreamPool &streams)
 
void update_values (Graph< T, S > *graph, StreamPool &streams)
 
template<typename I >
void build_csc_structure (Graph< T, S > *graph, CSCMatrix< S, I > &matrix)
 
template<typename I >
void update_csc_values (Graph< T, S > *graph, CSCMatrix< S, I > &matrix)
 

Public Attributes

std::unordered_map< BlockCoordinates, size_t > block_indices
 
thrust::device_vector< S > d_hessian
 
thrust::device_vector< size_t > d_col_pointers
 
thrust::device_vector< size_t > d_row_indices
 
thrust::device_vector< size_t > d_offsets
 
thrust::device_vector< S > d_prev_diagonal
 
thrust::device_vector< size_t > d_hessian_offsets
 
thrust::device_vector< size_t > scalar_to_block_map
 
thrust::host_vector< size_t > h_block_offsets
 
thrust::device_vector< size_t > d_block_offsets
 

The documentation for this class was generated from the following file: