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

Method dynamic_batch

mlx/data/Buffer.cpp:45–60  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

43}
44
45Buffer Buffer::dynamic_batch(
46 const std::string& key,
47 int64_t min_data_size,
48 int64_t max_data_size,
49 const std::unordered_map<std::string, double>& pad_values,
50 const std::unordered_map<std::string, int>& batch_dims,
51 bool drop_outliers) const {
52 return Buffer(std::make_shared<buffer::DynamicBatch>(
53 self_,
54 key,
55 min_data_size,
56 max_data_size,
57 pad_values,
58 batch_dims,
59 drop_outliers));
60}
61
62Buffer Buffer::dynamic_batch(
63 const Buffer& size_buffer,

Calls 1

BufferClass · 0.70