View Source Nx.Defn.Evaluator (Nx v0.5.2)
The default implementation of a Nx.Defn.Compiler
that evaluates the expression tree against the
tensor backend.
options
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.Servingwith this compiler