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

Method init_output_static_infer_desc

src/opr/impl/dnn/convolution.cpp:314–319  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

312}
313
314void ConvolutionForward::init_output_static_infer_desc() {
315 Super::set_nr_managed_outputs(this->output().size() - 1);
316 Super::init_output_static_infer_desc();
317 init_output_static_infer_desc_workspace(
318 intl::AutoAddWorkspaceNeedLimitGetter<megdnn::ConvolutionForward>::val);
319}
320
321void ConvolutionForward::get_output_var_shape(
322 const TensorShapeArray& inp_shape, TensorShapeArray& out_shape) const {

Callers

nothing calls this directly

Calls 4

sizeMethod · 0.45
outputMethod · 0.45

Tested by

no test coverage detected