Home
last modified time | relevance | path

Searched refs:depthwise_filter (Results 1 – 4 of 4) sorted by relevance

/dports/lang/halide/Halide-release_2019_08_27-2654-g664dc4993/apps/depthwise_separable_conv/
H A Dprocess.cpp17 Buffer<float> depthwise_filter(CM, CI, 3, 3); in main() local
31 for (int c = 0; c < depthwise_filter.dim(3).extent(); c++) { in main()
32 for (int z = 0; z < depthwise_filter.channels(); z++) { in main()
33 for (int y = 0; y < depthwise_filter.height(); y++) { in main()
34 for (int x = 0; x < depthwise_filter.width(); x++) { in main()
35 depthwise_filter(x, y, z, c) = rand(); in main()
57 depthwise_filter, in main()
68 depthwise_filter, in main()
H A Ddepthwise_separable_conv_generator.cpp14 Input<Buffer<float>> depthwise_filter{"depthwise_filter", 4}; member in __anon162e9f0a0111::DepthwiseSeparableConvolution
43 Expr channel_multiplier = depthwise_filter.dim(0).extent(); in generate()
48 Expr pad_width = depthwise_filter.dim(2).extent() / 2; in generate()
49 Expr pad_height = depthwise_filter.dim(3).extent() / 2; in generate()
50 RDom depthwise_filter_dom(0, depthwise_filter.dim(0).extent(), in generate()
51 0, depthwise_filter.dim(2).extent(), in generate()
58 depthwise_filter(rd, d, rx, ry) * in generate()
86 depthwise_filter.dim(0).set_estimate(0, CI / CO); in generate()
87 depthwise_filter.dim(1).set_estimate(0, CI); in generate()
88 depthwise_filter.dim(2).set_estimate(0, 3); in generate()
[all …]
H A Dtf_separable_conv.py7 depthwise_filter = tf.random.uniform([3, 3, 32, 1]) variable
17 img, depthwise_filter, pointwise_filter,
/dports/math/py-theano/Theano-1.0.5/theano/tensor/nnet/tests/
H A Dtest_abstract_conv.py1709 top = fun(self.x, self.depthwise_filter, self.pointwise_filter)
1713 top = fun(self.x[:, :, :3, :], self.depthwise_filter, self.pointwise_filter)
1725 top = fun(self.x, self.depthwise_filter, self.pointwise_filter)
1735 top = fun(self.x, self.depthwise_filter, self.pointwise_filter)
1749 depthwise_filter = np.tile(np.expand_dims(self.depthwise_filter, axis=2), (1, 1, 3, 1, 1))
1761 top = fun(x, depthwise_filter, pointwise_filter)
1764 top = fun(x[:, :, :3, :, :3], depthwise_filter, pointwise_filter)
1772 depthwise_filter_shape=depthwise_filter.shape,
1775 top = fun(x, depthwise_filter, pointwise_filter)
1785 top = fun(x, depthwise_filter, pointwise_filter)
[all …]