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