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

Method Convolution3DBackwardData

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

Source from the content-addressed store, hash-verified

575IMPL_CONV(Convolution3DBackwardData);
576
577Convolution3DBackwardData::Convolution3DBackwardData(
578 VarNode* filter, VarNode* diff, VarNode* src_for_shp, const Param& param,
579 const ExecutionPolicy& policy, const OperatorNodeConfig& config)
580 : Super{filter->owner_graph(), config, "conv3d_bwd_data", {filter, diff}} {
581 init_megdnn_opr(*this, param);
582 m_policy = policy;
583 add_input({filter, diff});
584 if (src_for_shp) {
585 add_input({src_for_shp});
586 }
587 output(0)->add_flag(VarNode::Flag::ALLOW_EMPTY_SHAPE);
588}
589
590SymbolVar Convolution3DBackwardData::make(
591 SymbolVar filter, SymbolVar diff, SymbolVar src, const Param& param,

Callers 1

conv_transpose3dFunction · 0.80

Calls 1

owner_graphMethod · 0.45

Tested by

no test coverage detected