本文整理汇总了C++中halide::Func::vectorize方法的典型用法代码示例。如果您正苦于以下问题：C++ Func::vectorize方法的具体用法？C++ Func::vectorize怎么用？C++ Func::vectorize使用的例子？那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类halide::Func的用法示例。
在下文中一共展示了Func::vectorize方法的3个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的C++代码示例。
示例1: convolution_layer
Halide::Func convolution_layer(Halide::Func input, Halide::Func weights,
Halide::Func bias, int filter_size, int input_layers, int pool_size) {
// Convolution
Halide::Func convolution;
Halide::Var x, y, z, w;
Halide::RDom r(0, filter_size, 0, filter_size, 0, input_layers);
convolution(x, y, z, w) = 0.0f;
convolution(x, y, z, w) += weights(r.x, r.y, r.z, z) *
input(x + r.x, y + r.y, r.z, w);
// Max pool
Halide::Func subsample;
Halide::RDom s(0, pool_size, 0, pool_size);
subsample(x, y, z, w) = 0.0f;
subsample(x, y, z, w) = Halide::max(convolution(pool_size * x + s.x,
pool_size * y + s.y, z, w), subsample(x, y, z, w));
// Non-linear bias
Halide::Func biased;
biased(x, y, z, w) = tanh(subsample(x, y, z, w) + bias(z, 0));
Halide::Var x_inner, x_outer, y_inner, y_outer;
biased.parallel(w);
biased.tile(x, y, x_outer, y_outer, x_inner, y_inner, VECTORS, 2);
biased.vectorize(x_inner);
biased.unroll(y_inner);
return biased;
}示例2: fully_connected_layer
Halide::Func fully_connected_layer(Halide::Func input, Halide::Func weights,
Halide::Func bias, int size) {
Halide::Func product;
Halide::Var x, y, z;
Halide::RDom r(0, size);
// Only y = 0 should be used
product(x, y, z) = 0.0f;
product(x, y, z) += weights(r.x, x) * input(r.x, y, z);
product(x, y, z) = tanh(product(x, y, z) + bias(x, 0));
product.vectorize(x, VECTORS);
return product;
}示例3: mat
void NamedWindow::showImage2D(Halide::Image<uint8_t> im)
{
static Halide::Func convert("convertToMat2D");
static Halide::ImageParam ip(Halide::UInt(8), 2);
static Halide::Var x, y;
if (!convert.defined())
{
convert(x, y) = ip(x, y);
convert.vectorize(x, 4).parallel(y, 4);
}
ip.set(im);
cv::Mat mat(im.height(), im.width(), CV_8UC1, cv::Scalar(0));
convert.realize(Halide::Buffer(Halide::UInt(8), im.width(), im.height(), 0, 0, mat.data));
cv::imshow(name, mat);
}