提交 f2579b1d authored 作者: Frederic's avatar Frederic

Add doc for python_constant_folding method.

上级 b60d4b35
...@@ -184,6 +184,20 @@ There are less methods to define for an Op than for a Type: ...@@ -184,6 +184,20 @@ There are less methods to define for an Op than for a Type:
Overrides :meth:`c_code_cache_version` if defined, but Overrides :meth:`c_code_cache_version` if defined, but
otherwise has the same contract. otherwise has the same contract.
.. method:: python_constant_folding(node)
Optional. If present this method will be called before doing
constant folding of a node, with that node as a parameter. If
it return True, we will not generate c code when doing constant
folding of this node. This is useful when the compilation of
the c code will be longer then the computation in python
(e.g. Elemwise of scalars).
In addition, this allow to lower the number of compiled module
and disk access. Particularly useful when the file system load
is high or when theano compilation directory is shared by many
process (like on a network file server on a cluster).
The ``name`` argument is currently given an invalid value, so steer The ``name`` argument is currently given an invalid value, so steer
away from it. As was the case with Type, ``sub['fail']`` provides away from it. As was the case with Type, ``sub['fail']`` provides
failure code that you *must* use if you want to raise an exception, failure code that you *must* use if you want to raise an exception,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论