HMLP: High-performance Machine Learning Primitives
Public Member Functions | Public Attributes | Static Public Attributes | List of all members
gnbx_mrxnr< MR, NR, OPKERNEL, OP1, OP2, TA, TB, TC, TPACKC, TV > Struct Template Reference

#include <fused_mrxnr.hpp>

Public Member Functions

void operator() (int k, TA *a, TB *b, TC *c, TV *v, int rs_v, int cs_v, aux_s< TA, TB, TC, TV > *aux) const
 

Public Attributes

OPKERNEL opkernel
 
OP1 op1
 
OP2 op2
 
TV initV
 

Static Public Attributes

static const size_t mr = MR
 
static const size_t nr = NR
 
static const size_t pack_mr = MR
 
static const size_t pack_nr = NR
 
static const size_t align_size = 32
 

Detailed Description

template<int MR, int NR, typename OPKERNEL, typename OP1, typename OP2, typename TA, typename TB, typename TC, typename TPACKC, typename TV>
struct gnbx_mrxnr< MR, NR, OPKERNEL, OP1, OP2, TA, TB, TC, TPACKC, TV >

end struct gkrm_mrxnr

Member Function Documentation

template<int MR, int NR, typename OPKERNEL, typename OP1, typename OP2, typename TA, typename TB, typename TC, typename TPACKC, typename TV>
void gnbx_mrxnr< MR, NR, OPKERNEL, OP1, OP2, TA, TB, TC, TPACKC, TV >::operator() ( int  k,
TA *  a,
TB *  b,
TC *  c,
TV *  v,
int  rs_v,
int  cs_v,
aux_s< TA, TB, TC, TV > *  aux 
) const
inline

Semiring rank-k update

Store back

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