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)
