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

Method get_workspace_in_bytes

dnn/src/naive/batch_conv_bias/opr_impl.cpp:44–48  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

42}
43
44size_t BatchConvBiasForwardImpl::get_workspace_in_bytes(
45 const TensorLayout& src, const TensorLayout& flt, const TensorLayout& bias,
46 const TensorLayout& z, const TensorLayout& dst) {
47 return get_workspace_bundle(nullptr, src, flt, bias, z, dst).total_size_in_bytes();
48}
49
50void BatchConvBiasForwardImpl::exec(
51 _megdnn_tensor_in src, _megdnn_tensor_in filter, _megdnn_tensor_in bias,

Callers

nothing calls this directly

Calls 2

total_size_in_bytesMethod · 0.80
get_workspace_bundleFunction · 0.50

Tested by

no test coverage detected