root
root copied to clipboard
[tmva][sofie] Updating BatchNormalization Op to handle paramteric input tensor
This Pull request:
Updates the implementation of BatchNormalization Operator to parse input tensors with parametric shapes.
xinput tensor can have shapes of form{dim1, dim2, dim3}scale,B,input_mean,input_vartensors need to be initialized tensors
Changes or fixes:
- Updated implementation in the
ROperator_BatchNormalization.hxxfile to handle parametric shapes.
Checklist:
- [x] tested changes locally
- [ ] updated the docs (if necessary)
This PR fixes #