Skip to content

Latest commit

 

History

History
27 lines (20 loc) · 1020 Bytes

README.md

File metadata and controls

27 lines (20 loc) · 1020 Bytes
Raw

Swift for TensorFlow Ops Bindings

This repository contains TensorFlow ops bindings for Swift for TensorFlow.

These bindings are automatically generated from TensorFlow ops specified either using ops registered to the TensorFlow runtime or via a protobuf file similar to ops.pbtxt in the main TensorFlow repo.

How to regenerate the bindings

To regenerate the swift ops bindings, run the following command. Note that this will use the TensorFlow (1.9 or above) python package.

python generate_wrappers.py --output_path=RawOpsGenerated.swift

Documentation gets automatically generated when adding a path to the api_def proto directory. This directory should contain per operator api_def protos with names like api_def_OpName.pbtxt.

python generate_wrappers.py --output_path=RawOpsGenerated.swift --api_def_path=/path/to/tensorflow/core/api_def/base_api