diff --git a/README.rst b/README.rst index 87a99e65e1b7814ab54da938780e746db6574ee6..17b2a588f1f1db6c7693f49daa5879834d206939 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 ------------------------