Commit 7aef4622 authored by James Keeling's avatar James Keeling Committed by TensorFlower Gardener
Browse files

Add GetFilteredRegisteredKernels and refactor

GetFilteredRegisteredKernels makes it easier for users to query at runtime which kernels are available which match some predicate. The most common usage will be querying which kernels are available for a given op, so we add the specialized GetRegisteredKernelsForOp.

This is part of the work to make available kernels possible to query, to support Swift For TensorFlow. There are also a number of github issues asking for the functionality.

I will add C API and Python API support in upcoming changes.

PiperOrigin-RevId: 205656251
parent 21d02059
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment