coremltools 6.1
- Support for TensorFlow
2.10. - New PyTorch ops supported:
baddbmm,glu,hstack,remainder,weight_norm,hann_window,randint,cross,trace, andreshape_as. - Avoid root logger and use the coremltools logger instead.
- Support dynamic input shapes for PyTorch
repeatandexpandop. - Enhance translation of torch
whereop with only one input. - Add support for PyTorch einsum equation:
'bhcq,bhck→bhqk’. - Optimization graph pass improvement
- 3D convolution batchnorm fusion
- Consecutive relu fusion
- Noop elimination
- Actively catch the tensor which has rank >= 6 and error out
- Various other bug fixes, optimizations and improvements.
Special thanks to our external contributors for this release: @fukatani, @piraka9011, @giorgiop, @hollance, @SangamSwadiK, @RobertRiachi, @waylybaye, @GaganNarula, and @sunnypurewal.