slicing.h File Reference

#include "mlx/array.h"

Go to the source code of this file.

Namespaces
namespace	mlx
namespace	mlx::core

Functions
void	mlx::core::slice_gpu (const array &in, array &out, const Shape &start_indices, const Shape &strides, const Stream &s)
void	mlx::core::concatenate_gpu (const std::vector< array > &inputs, array &out, int axis, const Stream &s)
void	mlx::core::pad_gpu (const array &in, const array &val, array &out, const std::vector< int > &axes, const Shape &low_pad_size, const Stream &s)