Commit 366dd0ea authored by Derek Murray's avatar Derek Murray Committed by TensorFlower Gardener
Browse files

Add a more efficient constructor for scalar `Tensor` objects in host memory. 

[tf.data] Optimize the creation of tensors in `tf.data.Dataset.range()`.

This change improves the range benchmark from 148.7 ns/element to 122.4 ns/element.

PiperOrigin-RevId: 220279090
parent 1e7b4ead
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