reduce.h File Reference

#include "mlx/backend/common/utils.h"

Go to the source code of this file.

Classes
struct	mlx::core::ReductionPlan

Namespaces
namespace	mlx
namespace	mlx::core

Enumerations
enum	mlx::core::ReductionOpType {   mlx::core::ContiguousAllReduce , mlx::core::ContiguousReduce , mlx::core::ContiguousStridedReduce , mlx::core::GeneralContiguousReduce ,   mlx::core::GeneralStridedReduce , mlx::core::GeneralReduce }

Functions
ReductionPlan	mlx::core::get_reduction_plan (const array &x, const std::vector< int > &axes)
std::pair< Shape, Strides >	mlx::core::shapes_without_reduction_axes (const array &x, const std::vector< int > &axes)