root icon indicating copy to clipboard operation
root copied to clipboard

[tmva][sofie] Updating BatchNormalization Op to handle paramteric input tensor

Open extint opened this issue 1 year ago • 0 comments

This Pull request:

Updates the implementation of BatchNormalization Operator to parse input tensors with parametric shapes.

  • x input tensor can have shapes of form {dim1, dim2, dim3}
  • scale, B, input_mean, input_var tensors need to be initialized tensors

ONNX Documentation

Changes or fixes:

  • Updated implementation in the ROperator_BatchNormalization.hxx file to handle parametric shapes.

Checklist:

  • [x] tested changes locally
  • [ ] updated the docs (if necessary)

This PR fixes #

extint avatar Aug 21 '24 15:08 extint