Commit 616a81ce authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Extended UnrollBatchMatMul to support tensor with dimension greater than 3.... 

Extended UnrollBatchMatMul to support tensor with dimension greater than 3. Augmented UnrollBatchMatMul to handle transpose_[ab].

PiperOrigin-RevId: 233867480
parent 2b5c02c2
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