menoh
menoh copied to clipboard
Operator requests
operator | requiring models |
---|---|
Sigmoid | YOLOv2, (NStep)LSTM |
Transpose | YOLOv2, Super resolution (https://github.com/pfnet-research/menoh/issues/56#issuecomment-410566250), GGNN (https://github.com/pfnet-research/menoh/issues/56#issuecomment-418930691) |
Reshape | YOLOv2 |
Mul | SSD, (NStep)LSTM |
Conv (dilation > 1) | SSD |
Upsample | YOLOv3*, FPN* |
Gather | GGNN (https://github.com/pfnet-research/menoh/issues/56#issuecomment-418930691) |
Identity | GGNN (https://github.com/pfnet-research/menoh/issues/56#issuecomment-418930691) |
Conv (group > 1) | MobileNet |
+1 for Transpose
to run super resolution example of MXNet.
I want following operators to run chainer-chemistry implementation of GGNN.
-
Gather
operator andint32
tensors (c.f. #41)- The use of them can be avoided if we input one-hot vectors instead of
int32
tensors, and I take this approach for ONNX file I attached to #89. Butint32
tensors is more concise and desirable.
- The use of them can be avoided if we input one-hot vectors instead of
-
Transpose
operator -
Identity
operator- Chainer's
copy
function is exported asIdentity
operator (c.f. https://github.com/chainer/onnx-chainer/pull/27 ). But the use ofIdentity
could eliminated if we transform ONNX file.
- Chainer's
@Hakuyume Sigmoid
, Transpose
, Mul
, Conv(dilation > 1)
are now available on mkldnn_with_generic_fallback
backend. Please check.