Date: Saturday, December 12, 2020 @ 11:43:11 Author: svenstaro Revision: 773269
Add simple test case Added: tensorflow/trunk/test.py Modified: tensorflow/trunk/PKGBUILD ----------+ PKGBUILD | 10 +++++----- test.py | 12 ++++++++++++ 2 files changed, 17 insertions(+), 5 deletions(-) Modified: PKGBUILD =================================================================== --- PKGBUILD 2020-12-12 11:41:46 UTC (rev 773268) +++ PKGBUILD 2020-12-12 11:43:11 UTC (rev 773269) @@ -7,7 +7,7 @@ pkgname=(tensorflow tensorflow-opt tensorflow-cuda tensorflow-opt-cuda python-tensorflow python-tensorflow-opt python-tensorflow-cuda python-tensorflow-opt-cuda) pkgver=2.4.0rc4 _pkgver=2.4.0-rc4 -pkgrel=1 +pkgrel=2 pkgdesc="Library for computation using data flow graphs for scalable machine learning" url="https://www.tensorflow.org/"; license=('APACHE') @@ -275,7 +275,7 @@ } package_python-tensorflow() { - depends+=(tensorflow python-termcolor python-astor python-gast python-numpy python-protobuf absl-py python-h5py python-keras-applications python-keras-preprocessing python-tensorflow-estimator python-opt_einsum python-astunparse python-pasta) + depends+=(tensorflow python-termcolor python-astor python-gast python-numpy python-protobuf absl-py python-h5py python-keras-applications python-keras-preprocessing python-tensorflow-estimator python-opt_einsum python-astunparse python-pasta python-flatbuffers) cd "${srcdir}"/tensorflow-${_pkgver} _python_package tmp @@ -283,7 +283,7 @@ package_python-tensorflow-opt() { pkgdesc="Library for computation using data flow graphs for scalable machine learning (with AVX2 CPU optimizations)" - depends+=(tensorflow-opt python-termcolor python-astor python-gast python-numpy python-protobuf absl-py python-h5py python-keras-applications python-keras-preprocessing python-tensorflow-estimator python-opt_einsum python-astunparse python-pasta) + depends+=(tensorflow-opt python-termcolor python-astor python-gast python-numpy python-protobuf absl-py python-h5py python-keras-applications python-keras-preprocessing python-tensorflow-estimator python-opt_einsum python-astunparse python-pasta python-flatbuffers) conflicts=(python-tensorflow) provides=(python-tensorflow) @@ -293,7 +293,7 @@ package_python-tensorflow-cuda() { pkgdesc="Library for computation using data flow graphs for scalable machine learning (with CUDA)" - depends+=(tensorflow-cuda python-termcolor python-astor python-gast python-numpy cuda cudnn python-pycuda python-protobuf absl-py nccl python-h5py python-keras-applications python-keras-preprocessing python-tensorflow-estimator python-opt_einsum python-astunparse python-pasta) + depends+=(tensorflow-cuda python-termcolor python-astor python-gast python-numpy cuda cudnn python-pycuda python-protobuf absl-py nccl python-h5py python-keras-applications python-keras-preprocessing python-tensorflow-estimator python-opt_einsum python-astunparse python-pasta python-flatbuffers) conflicts=(python-tensorflow) provides=(python-tensorflow) @@ -303,7 +303,7 @@ package_python-tensorflow-opt-cuda() { pkgdesc="Library for computation using data flow graphs for scalable machine learning (with CUDA and AVX2 CPU optimizations)" - depends+=(tensorflow-opt-cuda python-termcolor python-astor python-gast python-numpy cuda cudnn python-pycuda python-protobuf absl-py nccl python-h5py python-keras-applications python-keras-preprocessing python-tensorflow-estimator python-opt_einsum python-astunparse python-pasta) + depends+=(tensorflow-opt-cuda python-termcolor python-astor python-gast python-numpy cuda cudnn python-pycuda python-protobuf absl-py nccl python-h5py python-keras-applications python-keras-preprocessing python-tensorflow-estimator python-opt_einsum python-astunparse python-pasta python-flatbuffers) conflicts=(python-tensorflow) provides=(python-tensorflow python-tensorflow-cuda) Added: test.py =================================================================== --- test.py (rev 0) +++ test.py 2020-12-12 11:43:11 UTC (rev 773269) @@ -0,0 +1,12 @@ +#!/usr/bin/env python + +import tensorflow as tf + +with tf.device("/GPU:0"): + a = tf.random.normal([1, 2]) + + +def temp(x): + return tf.shape(x)[0] + +tf.autograph.to_graph(temp) Property changes on: tensorflow/trunk/test.py ___________________________________________________________________ Added: svn:executable ## -0,0 +1 ## +* \ No newline at end of property
