|
|
std::vector< BlockCoordinates > | get_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, const bool use_identity, StreamPool &streams) |
| |
|
void | setup_hessian_computation (Graph< T, S > *graph, StreamPool &streams) |
| |
|
void | execute_hessian_computation (Graph< T, S > *graph, StreamPool &streams) |
| |
|
const thrust::device_vector< size_t > & | get_block_col_pointers () const |
| |
|
const thrust::device_vector< size_t > & | get_block_row_indices () const |
| |
|
const thrust::device_vector< size_t > & | get_block_value_offsets () const |
| |
|
const thrust::device_vector< S > & | get_values () const |
| |
|
S * | get_values_ptr () |
| |
|
const S * | get_values_ptr () const |
| |
|
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) |
| |
|
|
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 |
| |