yobx.tensorflow.register#

yobx.tensorflow.register.get_tf_op_converter(op_type: str) Callable | None[source]#

Returns the converter for a TF op type, or None if not found.

yobx.tensorflow.register.get_tf_op_converters() Dict[str, Callable][source]#

Returns all registered TF op converters.

yobx.tensorflow.register.register_tf_op_converter(op_type: str | Tuple[str, ...])[source]#

Decorator that registers a converter for one or more TF op-type strings.