From 539f17aebf10dc5fabd55f17ae88c52f47a638cd Mon Sep 17 00:00:00 2001 From: Yong He Date: Fri, 31 Mar 2023 16:21:33 -0700 Subject: Update a1-02-slangpy.md --- docs/user-guide/a1-02-slangpy.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'docs/user-guide') diff --git a/docs/user-guide/a1-02-slangpy.md b/docs/user-guide/a1-02-slangpy.md index fff806229..4e7c8d0a4 100644 --- a/docs/user-guide/a1-02-slangpy.md +++ b/docs/user-guide/a1-02-slangpy.md @@ -242,10 +242,10 @@ float computeOutputPixel(TensorView input, uint2 pixelLoc) { // Skip out of bounds pixels. if (x < 0 || x >= width) continue; - if (y < 0 || y >= height) continue; for (int y = pixelLoc.y - 1; y <= pixelLoc.y + 1; y++) { + if (y < 0 || y >= height) continue; sumValue += input[x, y]; count++; } @@ -322,10 +322,10 @@ float computeOutputPixel( { // Skip out of bounds pixels. if (x < 0 || x >= width) continue; - if (y < 0 || y >= height) continue; for (int y = pixelLoc.y - 1; y <= pixelLoc.y + 1; y++) { + if (y < 0 || y >= height) continue; sumValue += getInputElement(input, inputGradToPropagateTo, uint2(x, y)); count++; } -- cgit v1.2.3