We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a796618 commit 277c12fCopy full SHA for 277c12f
nx/lib/nx/defn/compiler.ex
@@ -76,7 +76,7 @@ defmodule Nx.Defn.Compiler do
76
@doc """
77
Callback for compilation of a parallelizable computation.
78
79
- Its main purpose is to compile a function for a given `Nx.Defn.Shard.Mesh`.
+ Its main purpose is to compile a function for a given `Nx.Defn.Mesh`.
80
81
Receives an opaque `key` used for caching, a `mesh`, a list of `vars`
82
in `[vars]`, the function `fun` which builds a defn expression, a list of
nx/lib/nx/defn/shard/mesh.ex nx/lib/nx/defn/mesh.exnx/lib/nx/defn/shard/mesh.ex renamed to nx/lib/nx/defn/mesh.ex
0 commit comments