Join GitHub today
GitHub is home to over 50 million developers working together to host and review code, manage projects, and build software together.Sign up
ONNX: fix bug in export of ops involving torch.bool type #40006
When an op involves creating a tensor of a certain type (such as torch.ones(...)), the tracer creates a
However, when the ONNX exporter maps back the integer to torch type, 10 is mapped to bool, 9 is mapped to complex128 and 8 is mapped to complex64.