MCPcopy Create free account
hub / github.com/ml-explore/mlx-data / Shuffle

Method Shuffle

mlx/data/buffer/Shuffle.cpp:13–14  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

11namespace buffer {
12
13Shuffle::Shuffle(const std::shared_ptr<Buffer>& buffer)
14 : Perm(buffer, rand_perm_(buffer->size())) {}
15
16std::vector<int64_t> Shuffle::rand_perm_(int64_t size) {
17 auto state = core::get_state();

Callers

nothing calls this directly

Calls 1

sizeMethod · 0.45

Tested by

no test coverage detected