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

Public Member Functions

 Meta (const Tensor< CPUContext > &out_grad, int skip_dims, bool first_dim=true)
 
void observeOriginalInput (int, const Tensor< CPUContext > &, Tensor< CPUContext > *, int)
 
void appendGradShape (vector< TIndex > *output_shape)
 

Data Fields

TIndex block_size
 
vector< TIndex > block_shape
 
bool first_dim
 

Detailed Description

Definition at line 391 of file reducer_functors.h.


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