Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Enhance tvm infra for op attrs #15854

Merged
merged 1 commit into from
Aug 13, 2019
Merged

Enhance tvm infra for op attrs #15854

merged 1 commit into from
Aug 13, 2019

Conversation

hzfan
Copy link
Contributor

@hzfan hzfan commented Aug 12, 2019

Description

New tvm infra to support ops with attrs.

Changes

  • OpDef accepts 2 new kwargs:
    • attrs: the attrs to be appended in the function names
    • attrs_valid: to specify the valid attrs combination (like axis < ndim)

Comments

Thank @yzhliu for guidance and review

@yzhliu yzhliu merged commit 67daae7 into apache:master Aug 13, 2019
anirudhacharya pushed a commit to anirudhacharya/mxnet that referenced this pull request Aug 20, 2019
gyshi pushed a commit to gyshi/incubator-mxnet that referenced this pull request Aug 21, 2019
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants