Tensor Comprehensions
|
#include <halide2isl.h>
Public Attributes | |
Halide::Internal::Stmt | init |
Halide::Internal::Stmt | update |
std::vector< size_t > | dims |
Enumerate all reductions in a statement, by looking for the ReductionInit and ReductionUpdate markers inserted during lowering (see tc2halide.h). Each reduction object stores a reference to the init and the update statement, although the init statement is probably not strictly needed, and a list of reduction dimensions in the domain of the update statement.
std::vector<size_t> tc::halide2isl::Reduction::dims |
Halide::Internal::Stmt tc::halide2isl::Reduction::init |
Halide::Internal::Stmt tc::halide2isl::Reduction::update |