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 D | process.cpp | 17 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 D | depthwise_separable_conv_generator.cpp | 14 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 D | tf_separable_conv.py | 7 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 D | test_abstract_conv.py | 1709 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 …]
|