Commit 7d724a81 authored by Sean Papay's avatar Sean Papay Committed by Rasmus Munk Larsen
Browse files

Tensordot partial shape inference (#8452) 

* implemented partial shape inference for tensordot
tensordot should now correctly infer the partial shape of the product, whenever possible

* updated tensordot tests to test partial shape inference

* removed TODO note
parent 54932540
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment