Commit ae06b4e5 authored by James Ring's avatar James Ring Committed by TensorFlower Gardener
Browse files

Add draft of C BitcastOp implementation. 

This CL removes the C++ implementation of BitCast kernel and adds a C
implementation in a new tensorflow/c/kernels. The kernel is registered both for
CPU and, if CUDA is present, GPU.

The Bitcast Op registration is left in place. A C API for registering ops is
coming.

PiperOrigin-RevId: 237183873
parent a39f3150
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