MLX
 
Loading...
Searching...
No Matches
copy.h File Reference
#include <optional>
#include "mlx/array.h"
#include "mlx/backend/common/copy.h"
#include "mlx/backend/common/utils.h"

Go to the source code of this file.

Namespaces

namespace  mlx
 
namespace  mlx::core
 

Functions

void mlx::core::copy (const array &src, array &dst, CopyType ctype, Stream stream)
 
void mlx::core::copy_inplace (const array &src, array &dst, CopyType ctype, Stream stream)
 
void mlx::core::copy_inplace (const array &src, array &dst, const Shape &data_shape, const Strides &i_strides, const Strides &o_strides, int64_t i_offset, int64_t o_offset, CopyType ctype, Stream stream, const std::optional< array > &dynamic_i_offset=std::nullopt, const std::optional< array > &dynamic_o_offset=std::nullopt)