void getCanonicalSize(const Halide::Buffer<>& buffer, int* width, int* height,
int* channels, int* batch)
{
CV_Assert(buffer.dimensions() == 4);
*width = buffer.extent(0);
*height = buffer.extent(1);
*channels = buffer.extent(2);
*batch = buffer.extent(3);
}
void Closure::found_buffer_ref(const string &name, Type type,
bool read, bool written, Halide::Buffer<> image) {
if (!ignore.contains(name)) {
debug(3) << "Adding buffer " << name << " to closure\n";
Buffer &ref = buffers[name];
ref.type = type.element_of(); // TODO: Validate type is the same as existing refs?
ref.read = read;
ref.write = written;
// If reading an image/buffer, compute the size.
if (image.defined()) {
ref.size = image.size_in_bytes();
ref.dimensions = image.dimensions();
}
} else {
debug(3) << "Not adding " << name << " to closure\n";
}
}