apple / apple/coremltools

coreml using torch export API support maxpool

Open
#2,415 10 comments 0 reactions 1 assignee Claimed by @YifanShenSZ View on GitHub
feature request torch.export
Dominant language
Python
Stars
5.4k
Forks
850
Avg merge
4d 5h
Merged PRs (30d)
10

Description

## 🌱 Describe your Feature Request
coremltools export API failed to export maxpool, because MaxPool not adapt for "_parse_positional_args" and "_parse_keyword_args"
parse for argument as "_convolution"

## How can this feature be used?
maxpool is a very common op in convolution network. without this support some popular network cannot export successful

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.