Skip to content

GitLab

"lightx2v/models/runners/base_runner.py" did not exist on "0f408885d43b82569111ae6ec7c510635c501f33"
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
batch_norm_2d_test.onnx 249 Bytes