#include <algorithm>
#include <cstdint>
#include <functional>
#include <memory>
#include <vector>
#include "mlx/allocator.h"
#include "mlx/dtype.h"
#include "mlx/event.h"
Go to the source code of this file.
Classes | |
class | mlx::core::array |
struct | mlx::core::array::ArrayIterator |
struct | mlx::core::array::Data |
struct | mlx::core::array::Flags |
Namespaces | |
namespace | mlx |
namespace | mlx::core |
Typedefs | |
using | mlx::core::Deleter = std::function<void(allocator::Buffer)> |
using | mlx::core::ShapeElem = int32_t |
using | mlx::core::Shape = std::vector<ShapeElem> |
using | mlx::core::Strides = std::vector<int64_t> |
template<typename... T> | |
using | mlx::core::enable_for_arrays_t = typename std::enable_if_t<is_arrays_v<T...>> |
Variables | |
template<typename T> | |
constexpr bool | mlx::core::is_array_v |
template<typename... T> | |
constexpr bool | mlx::core::is_arrays_v = (is_array_v<T> && ...) |