coremltools
coremltools copied to clipboard
Enhance torch.zeros_like for GPU inference in dynamic shape
Add dtype support and unit test. The old PR #2345 is a wrong implementation.