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 std::vector< int > &start_indices, const std::vector< int > &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, std::vector< int > axes, std::vector< int > low_pad_size, const Stream &s)