Support fusing 3D Conv with Add/Mul.#23
Conversation
There was a problem hiding this comment.
Good catch. Thanks!
There was a problem hiding this comment.
Does this have the same problem as the conv_add_fusion where the other dims of the multiplier should be 1?
There was a problem hiding this comment.
Yeah, it is better to check it.
onnxruntime/core/graph/initializer.h
Outdated
There was a problem hiding this comment.
These helpers are a little weird with the broadcasting rules. mul now behaves different than add/sub/etc. Maybe just have a scale_by_scalar to handle the case for now?
There was a problem hiding this comment.
Agree. The PR is updated. Please review it! Thanks!
ca5452a to
66f89ce
Compare
With this PR, the subgraph 3D Conve->Add->Mul in Resnet3D can be fused into one 3D Conv.
4ac0998 to
5f87427
Compare
|
|
@weixingzhang its good for merge now |
…obuf_status Create a mapping from ONNX Runtime Status to Protobuf Status
With this PR, the subgraph 3D Conve->Add->Mul in Resnet3D can be fused into one 3D Conv.