Commit faeb0f7
committed
Update on "[ET-VK] Creating specialized version of conv2d pw shader for X and Y stride = 1 and padding = 0."
This diff creates a specialized version of the conv2d pw shader for X and Y stride equals 1 and padding equals 0.
* It adds a new file `conv2d_pw_s1p0.glsl`, which implements the conv2d pw shader for X and Y stride equals 1 and padding equals 0.
* It adds a new file `conv2d_pw_s1p0.yaml`, which defines the parameters and shader variants for the specialized conv2d pw shader.
* The file `Convolution.cpp` is modified to add a new parameter `stride_1_padding_0` to the `conv2d` function, which enables the use of the specialized shader.
Differential Revision: [D75423931](https://our.internmc.facebook.com/intern/diff/D75423931/)
[ghstack-poisoned]File tree
0 file changed
+0
-0
lines changed0 file changed
+0
-0
lines changed
0 commit comments