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

Function array_pad

mlx/data/Array.cpp:236–252  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

234
235template <class T>
236void array_pad(
237 void* dst,
238 void* src,
239 int64_t lchunksize,
240 int64_t chunksize,
241 int64_t rchunksize,
242 int64_t size,
243 int64_t itemsize) {
244 auto dst_t = reinterpret_cast<T*>(dst);
245 auto src_t = reinterpret_cast<T*>(src);
246 for (int64_t n = 0; n < size; n++) {
247 dst_t += lchunksize;
248 std::memcpy(dst_t, src_t, chunksize * itemsize);
249 dst_t += chunksize + rchunksize;
250 src_t += chunksize;
251 }
252}
253
254std::shared_ptr<Array> pad(
255 const std::shared_ptr<const Array>& arr,

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected