ONE icon indicating copy to clipboard operation
ONE copied to clipboard

Published 20 hours ago • verified publisher icon Samsung

[onert] NYI Shape Inference

Open hseok-oh opened this issue 3 years ago • 0 comments

https://github.com/Samsung/ONE/issues/8496#issuecomment-1050425440

I found our runtime don't support shape inference for DepthwiseConv2D yet. We need to support more operations for shape inference.

NYI static shape inference

  • [ ] AddN
  • [ ] BatchToSpaceND
  • [ ] DepthToSpace
  • [ ] DepthwiseConv2D
  • [ ] Einsum
  • [ ] EmbeddingLookup
  • [ ] HashtableLookup
  • [ ] InstanceNorm
  • [ ] LocalResponseNormalization
  • [ ] LogSoftmax
  • [ ] Pool2D
  • [ ] PReLU
  • [ ] Rank
  • [ ] ResizeNearestNeighbor
  • [ ] RNN
  • [ ] SpaceToDepth
  • [ ] SplitV
  • [ ] StatelessRandomUniform
  • [ ] TopKV2
  • [ ] TransposeConv

NYI dynamic shape inference

  • [ ] AddN
  • [ ] BatchToSpaceND
  • [ ] DepthToSpace
  • [ ] DepthwiseConv2D
  • [ ] Einsum
  • [ ] EmbeddingLookup
  • [ ] HashtableLookup
  • [ ] InstanceNorm
  • [ ] LocalResponseNormalization
  • [ ] LogSoftmax
  • [ ] Pool2D
  • [ ] PReLU
  • [ ] Rank
  • [ ] ResizeNearestNeighbor
  • [ ] RNN
  • [ ] SpaceToDepth
  • [ ] SplitV
  • [ ] StatelessRandomUniform
  • [ ] TopKV2
  • [ ] TransposeConv

hseok-oh avatar Feb 25 '22 01:02 hseok-oh