Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Charlie Lin's avatar
    Rewrite ONNX parse batch norm (#1362) · c00f8202
    Charlie Lin authored 
    Rewrites the BatchNormalization ONNX operator into other MIGX operators
- Added handling of 1D input tensor case (edge case in ONNX spec)
Removes the spatial and per_activation functionality (not in the ONNX spec)
- Did not remove the batch_norm_inference related code as the TensorFlow parser still uses it
- Can remove that code when the TF version is updated
    c00f8202
parse_batchnorm.cpp 4.75 KB