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 &out_grad, int skip_dims, bool first_dim=true)
 
void observeOriginalInput (int, const Tensor &, Tensor *, int)
 
void appendGradShape (vector< int64_t > *output_shape)
 

Data Fields

int64_t block_size
 
vector< int64_t > 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: