Skip to content

Commit

Permalink
Revert changes conv3d_naive_webgpu.ts
Browse files Browse the repository at this point in the history
  • Loading branch information
satyajandhyala committed Jul 23, 2024
1 parent 12685bc commit f6dd6ea
Showing 1 changed file with 4 additions and 8 deletions.
12 changes: 4 additions & 8 deletions js/web/lib/wasm/jsep/webgpu/ops/3rd-party/conv3d_naive_webgpu.ts
Original file line number Diff line number Diff line change
Expand Up @@ -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;
Expand Down

0 comments on commit f6dd6ea

Please sign in to comment.