Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
mvpant committed Jan 13, 2025
1 parent 252e218 commit 255b837
Showing 1 changed file with 11 additions and 11 deletions.
22 changes: 11 additions & 11 deletions stablehlo/testdata/bn_conv_fuse_float32.large.mlir
Original file line number Diff line number Diff line change
Expand Up @@ -16,18 +16,18 @@ module @jit_main attributes {torch.debug_module_name = "ResNet"} {
// Slicing the input to reduce CPU cycles spend in interpreter.
// Extracting a center of size 112x112 from an input of size 224x224.
//
// The adjustments of +4 for the height and +3 for the width are necessary
// The adjustments of +4 for the height and +3 for the width are necessary
// to account for the convolution configuration with [2, 2] stride and [3, 3] padding.
//
// 224 224
// 224
// ______________________________________
// | +4
// | ____
// | | X X X
// | +4 |
// | | X X X 112
// | | _______________
// | X X |X X X ... X|
// | +4
// | ____
// | | X X X
// | +4 |
// | | X X X 112
// | | _______________
// | X X |X X X ... X|
// 224 | | |
// | |X X X ... X|
// | | |
Expand All @@ -38,8 +38,8 @@ module @jit_main attributes {torch.debug_module_name = "ResNet"} {
// | | ... X|___
// | --------------- |
// | | +3
// | |
// |
// | |
// |
//
%input_slice = stablehlo.slice %0 [0:1, 0:3, 52:171, 52:171] : (tensor<1x3x224x224xf32>) -> tensor<1x3x119x119xf32>

Expand Down

0 comments on commit 255b837

Please sign in to comment.