Skip to content
Snippets Groups Projects
Commit 5b693423 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Rewrite tfl.batch_matmul with batch_1 constants of ones into reduce_sum, to... 

Rewrite tfl.batch_matmul with batch_1 constants of ones into reduce_sum, to avoid materializing the constant.

PiperOrigin-RevId: 613852834
parent bf2e6ff5
Hide whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment