Commit 75390d4c authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Special-case the AccumulateNV2 op in print_selective_registration_header 

AccumulateNV2 doesn't have or need a kernel. It gets rewritten to other ops by
accumulate_n_optimizer.cc. This change allows it to be mentioned in the output
of print_selective_registration_header, rather than being ignored with a
warning. Behavior for other ops is preserved.

PiperOrigin-RevId: 211701878
parent a3c1ccd1
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