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

Function copy_tensors

dnn/test/common/checker.cpp:98–108  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

96
97template <typename Impl>
98void copy_tensors(
99 const CheckerHelper::TensorValueArray& dest,
100 const CheckerHelper::TensorValueArray& src, const Impl& copy_impl) {
101 megdnn_assert(dest.size() == src.size(), "%zu != %zu", dest.size(), src.size());
102 for (size_t i = 0; i < src.size(); i++) {
103 auto&& tensor = src[i];
104 if (tensor.layout.ndim == 0)
105 continue;
106 memcpy_noncontig(dest[i].raw_ptr(), tensor.raw_ptr(), tensor.layout, copy_impl);
107 }
108}
109
110void copy_tensors(
111 const CheckerHelper::TensorValueArray& dest,

Callers 4

do_execMethod · 0.85
execMethod · 0.85

Calls 3

memcpy_noncontigFunction · 0.85
sizeMethod · 0.45
raw_ptrMethod · 0.45

Tested by

no test coverage detected