TensorFlow

Swift high-level API for TensorFlow.

49
2
Swift

TensorFlow Swift high-level API. Tweet

Swift 4.0 Platforms Linux & OS X GPL Octadero Twitter

Structure of API

architecture

API based on TensorFlow library.

  • CTensorFlow is C API system module;
  • CAPI - Swift writen low-level API to C library;
  • Proto - Swift auto - generated classes for TensorFlow structures and models;
  • OpPruducer - Swift writen command line tool to produce new TensorFlow Operations
  • TensorFlowKit - Swift writen high-level API;

Using library.

First of all you should install tensorflow_c library. You can do that using brew on mac os.
Version 1.4 currantly available for mac os and linux on Google cloud, so you can use it:
Also, you can install it from sources, how to install TensorFlow from sources you can find here.

Tutorials

Xcode

Make shure that you read README of submodule CTensorFlow

To generate xcode project file you can call:
If you install TensorFlow library (1.4.1 version) as brew package:

swift package -Xlinker -rpath -Xlinker /usr/local/Cellar/libtensorflow/1.4.1/lib/ generate-xcodeproj
swift package -Xlinker -rpath -Xlinker /server/repository/tensorflow/bazel-bin/tensorflow generate-xcodeproj

Where /server/repository/tensorflow/bazel-bin/tensorflow path to your TensorFlow C library /user/local/lib, usualy.

Also you can use config:

swift package generate-xcodeproj --xcconfig-overrides TensorFlow.xcconfig
  • At TensorFlow.xcconfig file set TENSORFLOW_PATH property with correct path.
  • It is important to set ‘TensorFlow.xcconfig’ name the same with projectfile.
  • There is issus SR-6073 with LD_RUNPATH_SEARCH_PATHS property. So, currently you have to set $(inherited) value manualy at your build variable.

Build and set RPATH setting:

#MacOS
swift build -Xlinker -rpath -Xlinker /usr/local/Cellar/libtensorflow/1.4.1/lib/
swift test -Xlinker -rpath -Xlinker /usr/local/Cellar/libtensorflow/1.4.1/lib/
#Linux
swift build -Xlinker -rpath -Xlinker /server/repository/tensorflow/bazel-bin/tensorflow
swift test -Xlinker -rpath -Xlinker /server/repository/tensorflow/bazel-bin/tensorflow

Features

Swift API provides accae to all available C features in TensorFlow library.

  • Create / read grapht;
  • Save statistic on file system;
  • Cunstruct and run session;
  • Include available operations;
  • Store and restore checkpoints by SavedModel class;

graph

Summary

Summary

Starting from version 0.0.5 you have posibility to track any metrics using TensorFlowKit.
That is easy way to visualize your model in Swift application.

You can visualize weights and biases:

summary-distribution

Draw your graph:

summary-graph

Track changes in 3D:

summary-histogram

Extract components as png images:

summary-image

Watch on dynamics of changes:

summary-scalar

Troubleshooting

  • In case app can’t load dynamic library on macOS:
otool -L libtensorflow_cc.so
sudo install_name_tool -id /%repository%/tensorflow/bazel-out/darwin_x86_64-opt/bin/tensorflow/libtensorflow_cc.so libtensorflow_cc.so
sudo install_name_tool -id /%repository%/tensorflow/bazel-out/darwin_x86_64-opt/bin/tensorflow/libtensorflow_framework.so libtensorflow_framework.so
sudo install_name_tool -id /%repository%/tensorflow/bazel-out/darwin_x86_64-opt/bin/tensorflow/libtensorflow.so libtensorflow.so
  • In case if app can’t load dynamic library on linux or mac os with error:
error while loading shared libraries: *libtensorflow_cc.so*: cannot open shared object file: No such file or directory

Add your library path linux:

export LD_LIBRARY_PATH=/server/repository/tensorflow/bazel-bin/tensorflow:/usr/local/lib/:$LD_LIBRARY_PATH

mac os:

export DYLD_LIBRARY_PATH=/server/repository/tensorflow/bazel-bin/tensorflow/:$DYLD_LIBRARY_PATH

C++ old related issues.

  • At build phase error:
warning: error while trying to use pkgConfig flags for CProtobuf: nonWhitelistedFlags("Non whitelisted flags found: [\"-D_THREAD_SAFE\", \"-D_THREAD_SAFE\"] in pc file protobuf")

Remove -D_THREAD_SAFE keys from file /usr/local/lib/pkgconfig/protobuf.pc

Error at build phase:

#include "tensorflow/core/framework/graph.pb.h"                                                                                                                                         ^                                                                                                                                                                     1 error generated.                                                                                                                                                             error: terminated(1): /usr/bin/swift-build-tool -f /server/repository/TensorFlow/.build/debug.yaml main

Download dependencies for C++ library at tensorflow repository.

tensorflow/contrib/makefile/download_dependencies.sh

Developing and extending API

Create new proto files

// Create temperory folder
mkdir /tmp/swift
cd %path-to-tensorflow-reposytory%

// Find all proto files and generate swift classes.
find. -name '*.proto' -print -exec protoc --swift_opt=Visibility=Public --swift_out=/tmp/swift {} \;

// All files will be removed after restart.
open /tmp/swift

Debuging

Sets the threshold for what messages will be logged.
Add TF_CPP_MIN_LOG_LEVEL=0 and TF_CPP_MIN_VLOG_LEVEL=0

List of operations

There are a few ways to get a list of the OpDefs for the registered ops:

TF_GetAllOpList in the C API retrieves all registered OpDef protocol messages. This can be used to write the generator in the client language. This requires that the client language have protocol buffer support in order to interpret the OpDef messages.

The C++ function OpRegistry::Global()->GetRegisteredOps() returns the same list of all registered OpDefs (defined in [tensorflow/core/framework/op.h]). This can be used to write the generator in C++ (particularly useful for languages that do not have protocol buffer support).

The ASCII-serialized version of that list is periodically checked in to [tensorflow/core/ops/ops.pbtxt] by an automated process.

OpProducer using C API to extract and prepare all available operation as Swift source code.