Skip to content
 
 

Latest commit

 

History

58 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

No description, website, or topics provided.

Resources

Contributing

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages