Modified to suppress broadcast when `Gemm` is immediately followed by a scalar `Mul`, `Div`, `Sub`, or `Mod`. - Githubissues

PINTO0309 / onnx2tf

Self-Created Tools to convert ONNX files (NCHW) to TensorFlow/TFLite/Keras format (NHWC). The purpose of this tool is to solve the massive Transpose extrapolation problem in onnx-tensorflow (onnx-tf). I don't need a Star, but give me a pull request.

MIT License

708 stars 73 forks source link

Modified to suppress broadcast when `Gemm` is immediately followed by a scalar `Mul`, `Div`, `Sub`, or `Mod`. #577

Closed PINTO0309 closed 10 months ago

PINTO0309 commented 10 months ago

1. Content and background

Mul, Div, Sub, or Mod
- Modified to suppress broadcast when Gemm is immediately followed by a scalar Mul, Div, Sub, or Mod.
- This is a very specific workaround for the special specifications of NNAPI.
- Address the issue of NNAPI not allowing multi-dimensional bias.
- The problem is not reproducible for variable batch sizes.
- The problem reproduces only when changing to a fixed batch size.

2. Summary of corrections

3. Before/After (If there is an operating log that can be used as a reference)

4. Issue number (only if there is a related issue)

pre_explicit_broadcast should not expend scalar tensor #573