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

Method PadToMultiple

mlx/data/op/Pad.cpp:65–71  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

63}
64
65PadToMultiple::PadToMultiple(
66 const std::string& ikey,
67 int dim,
68 int64_t size,
69 double value,
70 const std::string& okey)
71 : KeyTransformOp(ikey, okey), dim_(dim), size_(size), value_(value) {}
72std::shared_ptr<Array> PadToMultiple::apply_key(
73 const std::shared_ptr<const Array>& src) const {
74 auto dim = src->checkdim(dim_);

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected