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

Method get_workspace_in_bytes

dnn/src/naive/pooling/opr_impl.cpp:398–401  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

396};
397
398size_t PoolingForwardImpl::get_workspace_in_bytes(
399 const TensorLayout& src, const TensorLayout& dst) {
400 return get_workspace_bundle(nullptr, src, dst).total_size_in_bytes();
401}
402namespace {
403
404void post_process(const TensorND& dst, TensorND& comp_dst) {

Callers

nothing calls this directly

Calls 2

total_size_in_bytesMethod · 0.80
get_workspace_bundleFunction · 0.50

Tested by

no test coverage detected