msluszniak commited on
Commit
f21da1f
·
verified ·
1 Parent(s): f295c79

Core ML: emit a (1,520,520) int32 index map instead of (1,21,520,520) logits

Browse files

The argmax now runs inside the graph, built from reduce_max and elementwise ops so it stays on the ANE. 1.05x to 1.63x faster and 7 to 48 percent lower peak memory on an iPhone 16. XNNPACK variants are unchanged.

coreml/deeplab_v3_resnet101_coreml_fp16.pte CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4122979ac7ce7ab0b5a4e59859745813f7ffef4d7a15b3a7085271dca2c6e5ff
3
- size 117845264
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1e4f2883be246cebff4f84ad5b0def622dfa1a08ffaab1b4145e4155fdbb3f6b
3
+ size 117854661