mlprodict
mlprodict copied to clipboard
Support symbolic function with ShapeInference
The idea behind is to use a symbolic function to determine the output of every node assuming it depends on the input size. Module sympy can be used to do that. As a result, we could determine the derivative dShape(node i) / dN and check if this derivative is null. Then, it is easier to reduce memory footprint.