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