ONNX TensorFlow

flytekitplugins-onnxtensorflow

FlytekitModel Servingonnxtensorflowmodel

This plugin allows you to generate ONNX models from your TensorFlow Keras models.

Install

pip install flytekitplugins-onnxtensorflow

Quick Start(example, may need adjustment)

pip install flytekitplugins-onnxtensorflow

from flytekit import task, workflow
from flytekitplugins.onnxtensorflow import TensorFlow2ONNX, TensorFlow2ONNXConfig

config = TensorFlow2ONNX(...)

@task
def my_task() -> None:
    ...

@workflow
def my_workflow() -> None:
    my_task()

Available Imports (2)

configTensorFlow2ONNX

TensorFlow2ONNXConfig is the config used during the tensorflow to ONNX conversion.

extends dataclass — configuration or data structure for plugin setup

from flytekitplugins.onnxtensorflow import TensorFlow2ONNX

configTensorFlow2ONNXConfig

TensorFlow2ONNXConfig is the config used during the tensorflow to ONNX conversion.

extends dataclass — configuration or data structure for plugin setup

from flytekitplugins.onnxtensorflow import TensorFlow2ONNXConfig

Dependencies

tf2onnxtensorflow

Related Plugins

ONNX PyTorch

Model Serving

This plugin allows you to generate ONNX models from your PyTorch models.

ONNX ScikitLearn

Model Serving

This plugin allows you to generate ONNX models from your ScikitLearn models.

Dgxc-lepton

Model Serving

A professional Flytekit plugin that enables seamless deployment and management of AI inference endpoints using Lepton AI infrastructure within Flyte workflows.