MLX
Loading...
Searching...
No Matches
Namespaces | Enumerations | Functions
copy.h File Reference
#include "mlx/array.h"
#include "mlx/backend/common/utils.h"

Go to the source code of this file.

Namespaces

namespace  mlx
 
namespace  mlx::core
 

Enumerations

enum class  mlx::core::CopyType { mlx::core::Scalar , mlx::core::Vector , mlx::core::General , mlx::core::GeneralGeneral }
 

Functions

void mlx::core::copy (const array &src, array &dst, CopyType ctype)
 
void mlx::core::copy_inplace (const array &src, array &dst, CopyType ctype)
 
template<typename stride_t >
void mlx::core::copy_inplace (const array &src, array &dst, const std::vector< int > &data_shape, const std::vector< stride_t > &i_strides, const std::vector< stride_t > &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype)