1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
#include "Halide.h"
#include "daubechies_constants.h"
namespace {
Halide::Var x("x"), y("y"), c("c");
class haar_x : public Halide::Generator<haar_x> {
public:
Input<Buffer<float, 2>> in_{"in"};
Output<Buffer<float, 3>> out_{"out"};
void generate() {
Func in = Halide::BoundaryConditions::repeat_edge(in_);
out_(x, y, c) = mux(c,
{(in(2 * x, y) + in(2 * x + 1, y)),
(in(2 * x, y) - in(2 * x + 1, y))}) /
2;
out_.unroll(c, 2);
}
};
} // namespace
HALIDE_REGISTER_GENERATOR(haar_x, haar_x)
|
