[
https://issues.apache.org/jira/browse/SINGA-494?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
]
zhangzhaoqi updated SINGA-494:
------------------------------
Description:
Background: some autograd ops cannot satisfy the onnx demand, as following:
# *conv, averagepool, maxpool*
- only support 2d input, i.e, N*C*W*H
- not support SAME_UPPER, SAME_LOWER, count_include_pad and ceil_mod
# *reshape*
- not support zero_dim, zero_and_negative_dim
# *concat*
- not support 1d
# *matmul*
- only support 2d
# *min, max*
- only support 2 inputs
# *add*
- not support broadcast
# *and, or, xor*
- not support broadcast
# *div, pow, prelu*
- not support broadcast
Some improvements are being done.
was:
Background: some autograd ops cannot satisfy the onnx demand, as following:
# conv, averagepool, maxpool
- only support 2d input, i.e, N*C*W*H
- not support SAME_UPPER, SAME_LOWER, count_include_pad and ceil_mode
# reshape
- not support zero_dim, zero_and_negative_dim
# concat
- not support 1d
# matmul
- only support 2d
# min, max
- only support 2 inputs
# add
- not support broadcast
# and, or, xor
- not support broadcast
# div, pow, prelu
- not support broadcast
Some improvements are being done.
> Singa autograd improvement
> --------------------------
>
> Key: SINGA-494
> URL: https://issues.apache.org/jira/browse/SINGA-494
> Project: Singa
> Issue Type: New Feature
> Reporter: zhangzhaoqi
> Priority: Major
>
> Background: some autograd ops cannot satisfy the onnx demand, as following:
> # *conv, averagepool, maxpool*
> - only support 2d input, i.e, N*C*W*H
> - not support SAME_UPPER, SAME_LOWER, count_include_pad and ceil_mod
> # *reshape*
> - not support zero_dim, zero_and_negative_dim
> # *concat*
> - not support 1d
> # *matmul*
> - only support 2d
> # *min, max*
> - only support 2 inputs
> # *add*
> - not support broadcast
> # *and, or, xor*
> - not support broadcast
> # *div, pow, prelu*
> - not support broadcast
> Some improvements are being done.
--
This message was sent by Atlassian Jira
(v8.3.4#803005)
