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

Method DynamicBatch

mlx/data/buffer/DynamicBatch.cpp:15–31  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

13namespace buffer {
14
15DynamicBatch::DynamicBatch(
16 const std::shared_ptr<Buffer>& buffer,
17 const std::string& key,
18 int64_t min_data_size,
19 int64_t max_data_size,
20 const std::unordered_map<std::string, double>& pad_values,
21 const std::unordered_map<std::string, int>& batch_dims,
22 bool drop_outliers)
23 : DynamicBatch(
24 buffer,
25 nullptr,
26 key,
27 min_data_size,
28 max_data_size,
29 pad_values,
30 batch_dims,
31 drop_outliers) {};
32
33DynamicBatch::DynamicBatch(
34 const std::shared_ptr<Buffer>& buffer,

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected