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

Method init_output_static_infer_desc

src/opr/impl/dnn/pooling.cpp:33–38  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

31}
32
33void PoolingForward::init_output_static_infer_desc() {
34 Super::set_nr_managed_outputs(this->output().size() - 1);
35 Super::Super::init_output_static_infer_desc();
36 init_output_static_infer_desc_workspace(
37 intl::AutoAddWorkspaceNeedLimitGetter<megdnn::PoolingForward>::val);
38}
39
40size_t PoolingForward::get_workspace_size_bytes(
41 const TensorShapeArray& input_shapes,

Callers

nothing calls this directly

Calls 3

sizeMethod · 0.45
outputMethod · 0.45

Tested by

no test coverage detected