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

fix CreateOp of slice_channel to accept dtypes of uint8 and int32#9561

Merged
piiswrong merged 1 commit intoapache:masterfrom
ai-adv-lab:master
Jan 25, 2018
Merged

fix CreateOp of slice_channel to accept dtypes of uint8 and int32#9561
piiswrong merged 1 commit intoapache:masterfrom
ai-adv-lab:master

Conversation

@Soonhwan-Kwon
Copy link
Contributor

@Soonhwan-Kwon Soonhwan-Kwon commented Jan 25, 2018

Description

To deal with quantization in future,
operators like slice_channel(split) should have capability to accept uint8 and int32.
In slice_channel, it can be achieved by changing MSHADOW_REAL_TYPE_SWITCH to MSHADOW_TYPE_SWITCH as concat(https://github.com/apache/incubator-mxnet/blob/d2a856a3a2abb4e72edc301b8b821f0b75f30722/src/operator/concat.cu)
Is there any reason to use MSHADOW_REAL_TYPE_SWITCH for splitting symbol?

@piiswrong piiswrong merged commit 9aaa29d into apache:master Jan 25, 2018
rahul003 pushed a commit to rahul003/mxnet that referenced this pull request Jun 4, 2018
zheng-da pushed a commit to zheng-da/incubator-mxnet that referenced this pull request Jun 28, 2018
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants