Skip to content

Commit

Permalink
Small fixes in Resize CPU antialias (#19476)
Browse files Browse the repository at this point in the history
### Description
Add a comment, pass NCHW = true when setting upsample_antialias

### Motivation and Context
Small bugs.
  • Loading branch information
yuslepukhin authored Feb 9, 2024
1 parent 90cf037 commit 0e984ef
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion onnxruntime/core/providers/cpu/tensor/upsample_antialias.h
Original file line number Diff line number Diff line change
Expand Up @@ -700,7 +700,7 @@ void ResizeBiCubicAntiAlias(int64_t batch_size,
BiCubicParamsAntiAlias<typename AccumulateType<T>::type> p;
p.cubic_coeff_a = cubic_coeff_a;
SetupUpsampleFilterAntiAlias(p, input_paras, output_paras, scale_paras, roi,
alloc, get_original_coordinate, exclude_outside, false);
alloc, get_original_coordinate, exclude_outside, true);

return UpsampleBaseAntiAlias<T>(p, batch_size, num_channels, input_height, input_width, output_height, output_width,
use_extrapolation, extrapolation_value,
Expand Down

0 comments on commit 0e984ef

Please sign in to comment.