From 906de4b0c506646d42dfaf4aa31b0fa1770bf7e5 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Mon, 16 Dec 2019 15:25:58 +0100
Subject: [PATCH] Copy contents of index.rst to README.rst
---
README.rst | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/README.rst b/README.rst
index 87a99e6..17b2a58 100644
--- a/README.rst
+++ b/README.rst
@@ -92,13 +92,13 @@ You can also use the class `AutoDiffOp` to obtain both the assignments (if you a
op = AutoDiffOp(forward_assignments)
backward_assignments = op.backward_assignments
- tensorflow_op = op.create_tensorflow_op(backend='tensorflow_native')
+ tensorflow_op = op.create_tensorflow_op(backend='tensorflow_native', use_cuda=True)
... or Torch:
.. code-block:: python
- torch_op = op.create_tensorflow_op({x: x_tensor, y: y_tensor}, backend='torch_native')
+ torch_op = op.create_tensorflow_op(backend='torch_native', use_cuda=True)
Test Report and Coverage
------------------------
--
GitLab