From 3b6ec0fd06d08be1ad6162a9a881741ee6c97a9b Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Fri, 3 Jan 2020 13:16:21 +0100
Subject: [PATCH] Improve docstring of TensorflowModule
---
src/pystencils_autodiff/backends/astnodes.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/src/pystencils_autodiff/backends/astnodes.py b/src/pystencils_autodiff/backends/astnodes.py
index 95627c5..e51eb1b 100644
--- a/src/pystencils_autodiff/backends/astnodes.py
+++ b/src/pystencils_autodiff/backends/astnodes.py
@@ -141,8 +141,10 @@ class TensorflowModule(TorchModule):
def __init__(self, module_name, kernel_asts):
"""Create a C++ module with forward and optional backward_kernels
- :param forward_kernel_ast: one or more kernel ASTs (can have any C dialect)
- :param backward_kernel_ast:
+ Args:
+ module_name (str): Module name
+ kernel_asts (pystencils.kernel_wrappers.KernelWrapper):
+ ASTs as generated by `:func:pystencils.create_kernel`
"""
self.compiled_file = None
--
GitLab