Commit 87a3c967 authored by Akshay Agrawal's avatar Akshay Agrawal Committed by TensorFlower Gardener
Browse files

TFE: Register a GPU kernel for tfe.py_func. 

PiperOrigin-RevId: 183765122
parent e3d99c92
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