zhangzhaoqi created SINGA-494:
---------------------------------

             Summary: Singa autograd improvement
                 Key: SINGA-494
                 URL: https://issues.apache.org/jira/browse/SINGA-494
             Project: Singa
          Issue Type: New Feature
            Reporter: zhangzhaoqi


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.



--
This message was sent by Atlassian Jira
(v8.3.4#803005)

Reply via email to