Method getBuffer

mllm/nn/Module.cpp:74–74

void ModuleImpl::registerBuffer(const std::string& name, const Tensor& tensor) { buffer_.reg(name, tensor); }

Tensor ModuleImpl::getBuffer(const std::string& name) { return buffer_[name]; }

void ModuleImpl::updateBuffer(const std::string& name, const Tensor& tensor) { buffer_[name] = tensor; }