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
repeat
andexpand
op. - Enhance translation of torch
where
op 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.