Packages

nx

0.7.0
0.13.0 0.12.1 0.12.0 0.11.0 0.10.0 0.9.2 0.9.1 0.9.0 0.8.0 0.7.3 0.7.2 0.7.1 0.7.0 0.6.4 0.6.3 0.6.2 0.6.1 0.6.0 0.5.3 0.5.2 0.5.1 0.5.0 0.4.2 0.4.1 0.4.0 0.3.0 0.2.1 0.2.0 0.1.0

Multi-dimensional arrays (tensors) and numerical definitions for Elixir

HexDocs

Current section

Files

Jump to
Readme
29 Versions
2 Dependencies
Dependants
Files
Activity
nx lib nx defn debug.ex
Raw

lib/nx/defn/debug.ex 

defmodule Nx.Defn.Debug do

  @moduledoc false



  @behaviour Nx.Defn.Compiler



  @impl true

  def __partitions_options__(_), do: raise("not implemented")



  @impl true

  def __to_backend__(_), do: raise("not implemented")



  @impl true

  def __stream__(_, _, _, _, _, _, _), do: raise("not implemented")



  @impl true

  def __compile__(_, _, _, _), do: raise("not implemented")



  @impl true

  def __jit__(key, vars, fun, args, _opts) do

    Process.put(__MODULE__, key)

    expr = fun.(vars)

    Enum.map(args, fn _ -> expr end)

  end

end