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

Pin ops with small integer inputs (already on the cpu) to the cpu in eager. 

An environment variable (TF_EAGER_ENABLE_SMALL_TENSOR_CPU_PINNING) is provided to turn this off if necessary (its on by default).

PiperOrigin-RevId: 215821915
parent d6a2e7bc
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