MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / default_tensor_value_loader

Method default_tensor_value_loader

src/serialization/impl/opr_load_dump.cpp:75–83  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

73}
74
75void GraphLoadConfig::default_tensor_value_loader(
76 void* ptr, const TensorLayout& layout, InputFile& fin) {
77 auto sz = layout.span().high_byte;
78 if (ptr) {
79 fin.read(ptr, sz);
80 } else {
81 fin.skip(sz);
82 }
83}
84
85SharedBuffer OprLoadContextRawPOD::load_shared_buf(size_t size) {
86 std::shared_ptr<uint8_t> shptr{new uint8_t[size], [](uint8_t* p) { delete[] p; }};

Callers

nothing calls this directly

Calls 3

spanMethod · 0.80
readMethod · 0.45
skipMethod · 0.45

Tested by

no test coverage detected