mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-25 18:11:15 +08:00
17 lines
370 B
C
17 lines
370 B
C
|
// Copyright © 2024 Apple Inc.
|
||
|
|
|
||
|
|
#include <optional>
|
||
|
|
|
||
|
|
#include <nanobind/nanobind.h>
|
||
|
|
#include <nanobind/ndarray.h>
|
||
|
|
|
||
|
|
#include "mlx/array.h"
|
||
|
|
|
||
|
|
namespace nb = nanobind;
|
||
|
|
using namespace mlx::core;
|
||
|
|
|
||
|
|
array nd_array_to_mlx(
|
||
|
|
nb::ndarray<nb::ro, nb::c_contig, nb::device::cpu> nd_array,
|
||
|
|
std::optional<Dtype> dtype);
|
||
|
|
nb::ndarray<nb::numpy> mlx_to_np_array(const array& a);
|