Google Git
Sign in
android / platform / external / tensorflow / 119303cbdd1aeddbff4c47032b15de51fd9146af / . / tensorflow / go
tree: 22dddf0fd387edad60bcbcd5f644afe2ed510710 [path history] [tgz]
  1. genop/
  2. op/
  3. android.go
  4. attrs.go
  5. attrs_test.go
  6. BUILD
  7. context.go
  8. context_test.go
  9. doc.go
  10. example_inception_inference_test.go
  11. graph.go
  12. graph_test.go
  13. lib.go
  14. operation.go
  15. operation_test.go
  16. README.md
  17. saved_model.go
  18. saved_model_test.go
  19. session.go
  20. session_test.go
  21. shape.go
  22. shape_test.go
  23. status.go
  24. tensor.go
  25. tensor_handle.go
  26. tensor_handle_test.go
  27. tensor_test.go
  28. test.sh
  29. util_test.go
  30. version.go
tensorflow/go/README.md

TensorFlow in Go

Construct and execute TensorFlow graphs in Go.

GoDoc

WARNING: The API defined in this package is not stable and can change without notice. The same goes for the package path: (github.com/tensorflow/tensorflow/tensorflow/go).

Quickstart

Refer to Installing TensorFlow for Go

Building the TensorFlow C library from source

If the “Quickstart” instructions above do not work (perhaps the release archives are not available for your operating system or architecture, or you're using a different version of CUDA/cuDNN), then the TensorFlow C library must be built from source.

Prerequisites

  • bazel

  • Environment to build TensorFlow from source code (Linux or macOS). If you don't need GPU support, then try the following:

    sudo apt-get install python swig python-numpy # Linux
brew install swig                             # OS X with homebrew

Build

  1. Download the source code

    go get -d github.com/tensorflow/tensorflow/tensorflow/go

  2. Build the TensorFlow C library:

    cd ${GOPATH}/src/github.com/tensorflow/tensorflow
./configure
bazel build -c opt //tensorflow:libtensorflow.so

    This can take a while (tens of minutes, more if also building for GPU).

  3. Make libtensorflow.so and libtensorflow_framework.so available to the linker. This can be done by either:

    a. Copying it to a system location, e.g.,

    sudo cp ${GOPATH}/src/github.com/tensorflow/tensorflow/bazel-bin/tensorflow/libtensorflow.so /usr/local/lib
sudo cp ${GOPATH}/src/github.com/tensorflow/tensorflow/bazel-bin/tensorflow/libtensorflow_framework.so /usr/local/lib

    OR

    b. Setting environment variables:

    export LIBRARY_PATH=${GOPATH}/src/github.com/tensorflow/tensorflow/bazel-bin/tensorflow
# Linux
export LD_LIBRARY_PATH=${GOPATH}/src/github.com/tensorflow/tensorflow/bazel-bin/tensorflow
# OS X
export DYLD_LIBRARY_PATH=${GOPATH}/src/github.com/tensorflow/tensorflow/bazel-bin/tensorflow

  4. Build and test:

    go test github.com/tensorflow/tensorflow/tensorflow/go

Generate wrapper functions for ops

Go functions corresponding to TensorFlow operations are generated in op/wrappers.go. To regenerate them:

Prerequisites:

go generate github.com/tensorflow/tensorflow/tensorflow/go/op

Support

Use stackoverflow and/or Github issues.

Contributions

Contributions are welcome. If making any signification changes, probably best to discuss on a Github issue before investing too much time. Github pull requests are used for contributions.