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 a Nx.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 to inspect/2 for the result and arguments of each node.