Commit 7d8bfd88 authored by Smit Hinsu's avatar Smit Hinsu Committed by TensorFlower Gardener
Browse files

Set kIntsonDeviceAttr for native function in TRTEngineOp 

This is required to make sure that all TRTEngineOp outputs are always on a GPU
device as expected. Otherwise, some of the tensors of type DT_INT32 would be on
host if the op generating the tensor has host memory tag set.

PiperOrigin-RevId: 231707027
parent 4a70bff9
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