View Source Nx.Defn.Evaluator (Nx v0.10.0)
The default implementation of a Nx.Defn.Compiler
that evaluates the expression tree against the
tensor backend.
Options
The following options are specific to this compiler:
:garbage_collect
- when true, garbage collects after evaluating each node:max_concurrency
- the number of partitions to start when running aNx.Serving
with this compiler:debug_options
- a keyword list of options for debugging the evaluation of the expression tree. If not given, no debugging information will be printed or saved. The following options are supported::save_path
- the base path for output files. If not given, the output will be printed to the standard output.:inspect_limit
- limit that will be passed toinspect/2
for the result and arguments of each node.