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 |
Definition at line 391 of file reducer_functors.h.