saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-23 22:13:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[OneDNN] Fix poolgrad bug (#15557)

Browse source 
* Fixed default dilatation value for poolgrad ops

### Description
Changed default dilatation value to 0 in poolgrad ops



### Motivation and Context
Fixes error on unit tests when --enable_training --use_dnnl flags are
active and
This commit is contained in:
Erick Muñoz 2023-04-23 08:20:26 -07:00 committed by GitHub
parent d1354dcc83
commit 45c82eefb4
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/core/providers/dnnl/subgraph/dnnl_poolgrad.cc
View file

@ -92,8 +92,8 @@ void DnnlPoolGrad::CreatePrimitive(DnnlSubgraphPrimitive& sp, DnnlNode& node) {
}
}
// Dilatation of 1
auto dilatation = dnnl::memory::dims(kernel_shape.size(), 1);
// Default dilatation to 0
auto dilatation = dnnl::memory::dims(kernel_shape.size(), 0);
dnnl::pooling_forward::primitive_desc pool_forward_pd(dnnl_engine, dnnl::prop_kind::forward, algo, fwd_dx_md, dy_md,
strides, kernel_shape, dilatation, padding_left, padding_right);