TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.load_op_library

TensorFlow 1 version View source on GitHub

Loads a TensorFlow plugin, containing custom ops and kernels.

Aliases:

tf.load_op_library(library_filename)

Pass "library_filename" to a platform-specific mechanism for dynamically loading a library. The rules for determining the exact location of the library are platform-specific and are not documented here. When the library is loaded, ops and kernels registered in the library via the REGISTER_* macros are made available in the TensorFlow process. Note that ops with the same name as an existing op are rejected and not registered with the process.

Args:

  • library_filename: Path to the plugin. Relative or absolute filesystem path to a dynamic library file.

Returns:

A python module containing the Python wrappers for Ops defined in the plugin.

Raises:

  • RuntimeError: when unable to load the library or get the python wrappers.