本文整理汇总了C++中halide::Func::unroll方法的典型用法代码示例。如果您正苦于以下问题：C++ Func::unroll方法的具体用法？C++ Func::unroll怎么用？C++ Func::unroll使用的例子？那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类halide::Func的用法示例。
在下文中一共展示了Func::unroll方法的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的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;
}