Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
caffe2::WeightedSumReducer< T, CPUContext >::Meta Struct Reference
Inheritance diagram for caffe2::WeightedSumReducer< T, CPUContext >::Meta:
caffe2::BaseReducer::Meta

Public Member Functions

 Meta (bool first=true)
 
void observeInput (int input, const Tensor< CPUContext > &value, int skip_dims)
 
- Public Member Functions inherited from caffe2::BaseReducer::Meta
 Meta (bool first=true)
 
void computeMeta (const std::vector< TIndex > &dims, int skip_dims)
 
void observeInput (int input, const Tensor< CPUContext > &value, int skip_dims)
 
void appendOutputShape (vector< TIndex > *output_shape)
 
vector< TIndex > getOutputShape (const TensorShape &in, int skip_dims)
 

Data Fields

const T * scalars
 
bool first_dim
 
- Data Fields inherited from caffe2::BaseReducer::Meta
TIndex block_size
 
vector< TIndex > block_shape
 
bool first_dim
 

Detailed Description

template<typename T>
struct caffe2::WeightedSumReducer< T, CPUContext >::Meta

Definition at line 533 of file reducer_functors.h.


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