Mathieu Poumeyrol
Mathieu Poumeyrol
Ok this looks more serious, you can use "--pass analyse" to see the network before it crashes. it looks like the "then" and "else" branch have a different rank here,...
Ha. And we don't see the then and else branch, it's not plugged in. Can you link me the onnx file again so I can have a look ?
Please have a look at https://github.com/sonos/tract/pull/1102 . It's... complicated, but I guess the network is. I hope you can figure out where the discrep is.
The PR just fixes the dump format so that If branch are shown in the graph.
Wow. Different shapes? ONNX spec is borderline crazy here. I can manage to support different shapes of the same rank with symbols, but supporting different ranks is an absolute no...
What kind of shapes is netron reporting ? I am surprised that the convolution can deal with the two axes configuration, as its weights have to be of the same...
It may also be a tract bug while inferring shapes in one of the branches. Let's hope the Silero team can help us understanding what we are looking at.
Guess dropping the "sonos" name in the issue at Silero generated some confusion and/or wrong expectations :) Not sure we're gonna get much help from there.
Yeah, this is right, rust just does not have f16 support, so we must do it by hand. Assembly is a way, and we have infrastructure in linalg for this...
As you noticed, beyond a very relevant addition, this is also a sandbox for other operations that could benefit from having platform specific and/or optimized code. So just want to...