onnxruntime
f9cf6291 - Add same_shape case for BiasDropout (#9188)

Commit
4 years ago
Add same_shape case for BiasDropout (#9188) * bias dropout improvement * add transform case for same shape case * combine kernel * merge with vectorized kernel * use "has_same_shape_bias" * minor: a "N % 4 != 0" case * add op UT for has_same_shape_bias * address comments; add param case for 1d bias; add param case tests for 1d and same-shape bias * rewrite logic condition Co-authored-by: Peng Wang <pengwa@microsoft.com>
Author
Parents
Loading