Skip to content

model.fix_slice_nodes_for_onnx

Full path: schola.core.model.fix_slice_nodes_for_onnx

Fix Slice nodes produced by torch.onnx.export that use invalid 0-D tensor inputs.

fix_slice_nodes_for_onnx(model)

Parameters

  • model (onnx_ir.ir.Model) - ONNX IR model to mutate in place.