Commit 892ea4dc authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Add annotation to eager mode. 

With these annotations, when user turns on tracing, it will display op name like "BiasAddGrad" instead of kernel name "void tensorflow::BiasGradNCHW_SharedAtomics<float>(float const*, float*, int, int, int, int)".

PiperOrigin-RevId: 228951394
parent 67c18c4f
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