AlexanderLutsenko/nobuco

Add support for torch.repeat_interleave

stakemura opened this issue · 2 comments

stakemura commented

I really appreciate your great work.

Secondly, I found out that current conversion does not support torch.repeat_interleave .
It seems that tf.repeat match repeat_interleave.

I would be grateful if repeat_interleave conversion would be supported officially.

AlexanderLutsenko commented

@stakemura Hey, thank you! Added repeat_interleave in v0.10.2

nobuco/nobuco/node_converters/tensor_manipulation.py

Line 173 in 15673c2

@converter(torch.repeat_interleave, torch.Tensor.repeat_interleave, channel_ordering_strategy=ChannelOrderingStrategy.MINIMUM_TRANSPOSITIONS)

  • 👍1
stakemura commented

I confirmed that v.0.10.2 can convert the model including torch.repeat_interleave.
I truly appreciate your quick reply!

  • 🎉1