Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Types | Public Member Functions | Data Fields
at::native::WelfordOps< scalar_t, acc_scalar_t, index_t, combine_t > Struct Template Reference

Public Types

using acc_t = WelfordData< acc_scalar_t, index_t, combine_t >
 

Public Member Functions

C10_DEVICE acc_t reduce (acc_t acc, scalar_t data) const
 
C10_DEVICE acc_t combine (acc_t a, acc_t b) const
 
C10_DEVICE scalar_t project (acc_t acc) const
 
 WelfordOps (bool unbiased, bool take_sqrt)
 

Data Fields

bool unbiased
 
bool take_sqrt
 

Detailed Description

template<typename scalar_t, typename acc_scalar_t, typename index_t, typename combine_t>
struct at::native::WelfordOps< scalar_t, acc_scalar_t, index_t, combine_t >

Definition at line 46 of file SharedReduceOps.h.


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