Commit 625609c3 authored by Peter Hawkins's avatar Peter Hawkins Committed by TensorFlower Gardener
Browse files

[TF:XLA] Don't register kernels with infeasible type constraints. 

tensorflow::KernelAttrsMatch returns an error if it sees a KernelDef with an empty type constraint, and it is perfectly reasonable for an op (here TopK with a bfloat16 type) to only be supported on a subset of XLA backends.
PiperOrigin-RevId: 202144682
parent 06a318fe
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