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

Public Member Functions

 Meta (bool first=true)
void computeMeta (at::IntArrayRef dims, int skip_dims)
void observeInput (int input, const Tensor &value, int skip_dims)
void appendOutputShape (vector< int64_t > *output_shape)
vector< int64_t > getOutputShape (const TensorShape &in, int skip_dims)

Data Fields

int64_t block_size
vector< int64_t > block_shape
bool first_dim

Detailed Description

Definition at line 331 of file reducer_functors.h.

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