Commit 745b6aa4 authored by Akshay Modi's avatar Akshay Modi Committed by TensorFlower Gardener
Browse files

Use array_ops.split instead of slicing the tensor out individually. 

There is a speed difference in eager mode (I didn't measure functions)

PiperOrigin-RevId: 237485000
parent 14c73cc3
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