From f24200db2c3f2ad2e7fb6c91bbac741564165359 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?V=C3=ADctor=20Aguilar?= Date: Fri, 15 Dec 2023 13:46:50 -0800 Subject: [PATCH] accross -> across (#183) --- docs/src/dev/extensions.rst | 2 +- examples/extensions/axpby/axpby.h | 2 +- mlx/primitives.h | 2 +- python/mlx/nn/layers/normalization.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/docs/src/dev/extensions.rst b/docs/src/dev/extensions.rst index 9482be725..9aae931a3 100644 --- a/docs/src/dev/extensions.rst +++ b/docs/src/dev/extensions.rst @@ -150,7 +150,7 @@ back and go to our example to give ourselves a more concrete image. const std::vector& argnums) override; /** - * The primitive must know how to vectorize itself accross + * The primitive must know how to vectorize itself across * the given axes. The output is a pair containing the array * representing the vectorized computation and the axis which * corresponds to the output vectorized dimension. diff --git a/examples/extensions/axpby/axpby.h b/examples/extensions/axpby/axpby.h index 8c25980f4..9ff6af0b1 100644 --- a/examples/extensions/axpby/axpby.h +++ b/examples/extensions/axpby/axpby.h @@ -58,7 +58,7 @@ class Axpby : public Primitive { const std::vector& argnums) override; /** - * The primitive must know how to vectorize itself accross + * The primitive must know how to vectorize itself across * the given axes. The output is a pair containing the array * representing the vectorized computation and the axis which * corresponds to the output vectorized dimension. diff --git a/mlx/primitives.h b/mlx/primitives.h index c70de50d8..8141d0fca 100644 --- a/mlx/primitives.h +++ b/mlx/primitives.h @@ -72,7 +72,7 @@ class Primitive { const std::vector& argnums); /** - * The primitive must know how to vectorize itself accross + * The primitive must know how to vectorize itself across * the given axes. The output is a pair containing the array * representing the vectorized computation and the axis which * corresponds to the output vectorized dimension. diff --git a/python/mlx/nn/layers/normalization.py b/python/mlx/nn/layers/normalization.py index 348a8c545..6de377cda 100644 --- a/python/mlx/nn/layers/normalization.py +++ b/python/mlx/nn/layers/normalization.py @@ -97,7 +97,7 @@ class GroupNorm(Module): where :math:`\gamma` and :math:`\beta` are learned per feature dimension parameters initialized at 1 and 0 respectively. However, the mean and variance are computed over the spatial dimensions and each group of - features. In particular, the input is split into num_groups accross the + features. In particular, the input is split into num_groups across the feature dimension. The feature dimension is assumed to be the last dimension and the dimensions