From 923568c458a2e2df525cfca5858cfb0ca8b93977 Mon Sep 17 00:00:00 2001 From: Satya Jandhyala Date: Mon, 22 Jul 2024 16:56:26 -0700 Subject: [PATCH] Revert changes conv3d_naive_webgpu.ts --- .../jsep/webgpu/ops/3rd-party/conv3d_naive_webgpu.ts | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/js/web/lib/wasm/jsep/webgpu/ops/3rd-party/conv3d_naive_webgpu.ts b/js/web/lib/wasm/jsep/webgpu/ops/3rd-party/conv3d_naive_webgpu.ts index 26e6ec0f46bc6..f428293add599 100644 --- a/js/web/lib/wasm/jsep/webgpu/ops/3rd-party/conv3d_naive_webgpu.ts +++ b/js/web/lib/wasm/jsep/webgpu/ops/3rd-party/conv3d_naive_webgpu.ts @@ -298,17 +298,13 @@ export const createConv3DNaiveProgramInfo = let xRCorner = xFRCCorner.y; let xCCorner = xFRCCorner.z; let xShapeY = ${ - isChannelsLast ? getElementAt('uniforms.x_shape', 1, x.rank) : - getElementAt('uniforms.x_shape', 2, x.rank)}; + isChannelsLast ? getElementAt('uniforms.x_shape', 1, x.rank) : getElementAt('uniforms.x_shape', 2, x.rank)}; let xShapeZ = ${ - isChannelsLast ? getElementAt('uniforms.x_shape', 2, x.rank) : - getElementAt('uniforms.x_shape', 3, x.rank)}; + isChannelsLast ? getElementAt('uniforms.x_shape', 2, x.rank) : getElementAt('uniforms.x_shape', 3, x.rank)}; let xShapeW = ${ - isChannelsLast ? getElementAt('uniforms.x_shape', 3, x.rank) : - getElementAt('uniforms.x_shape', 4, x.rank)}; + isChannelsLast ? getElementAt('uniforms.x_shape', 3, x.rank) : getElementAt('uniforms.x_shape', 4, x.rank)}; let xShapeU = ${ - isChannelsLast ? getElementAt('uniforms.x_shape', 4, x.rank) : - getElementAt('uniforms.x_shape', 1, x.rank)}; + isChannelsLast ? getElementAt('uniforms.x_shape', 4, x.rank) : getElementAt('uniforms.x_shape', 1, x.rank)}; let inputDepthNearestVec4 = (xShapeU / 4) * 4; let inputDepthVec4Remainder = xShapeU % 4;