Commit 725206e6 authored by Peter Hawkins's avatar Peter Hawkins Committed by TensorFlower Gardener
Browse files

[TF:XLA] Register the _HostCast operator on XlaDevice subclasses. 

Declare CpuCastOp and CastOpBase in the cast_op.h header so they can be used from XlaDevice.

PiperOrigin-RevId: 170121111
parent 122ad249
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