"src/nni_manager/vscode:/vscode.git/clone" did not exist on "98f66f76d310b0e0679823d966fdaa6adafb66c2"
  • 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