Commit 6027fdf2 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Add python wrapper for xla::Einsum in tf2xla 

Using xla::Einsum is more efficient in some cases as it dosn't do
any unnecessary reshapes.

PiperOrigin-RevId: 238029298
parent 95b536f1
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