Commit 9ab9d5e0 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

In constant_op.cc get the CPU RAM allocator through the OpKernelContext 

instead of calling cpu_allocator() directly.  This will ensure that
NUMA node specific constant ops get node-local memory.

PiperOrigin-RevId: 202576292
parent 4740e1b6
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