Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
at::native::MeanOps< acc_t, factor_t > Struct Template Reference

Public Member Functions

C10_DEVICE acc_t reduce (acc_t a, acc_t b) const
 
C10_DEVICE acc_t combine (acc_t a, acc_t b) const
 
C10_DEVICE acc_t project (acc_t a) const
 
 MeanOps (factor_t factor)
 

Data Fields

factor_t factor
 

Detailed Description

template<typename acc_t, typename factor_t>
struct at::native::MeanOps< acc_t, factor_t >

Definition at line 106 of file SharedReduceOps.h.


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