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

Method SizeArgs

dnn/src/cambricon/conv_bias/algo.cpp:44–49  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

42using NonlineMode = param::ConvBias::NonlineMode;
43
44ConvBiasForwardImpl::AlgoBase::SizeArgs::SizeArgs(
45 ConvBiasForwardImpl* opr, const TensorLayout& src, const TensorLayout& filter,
46 const TensorLayout& bias, const TensorLayout& z, const TensorLayout& dst)
47 : SizeArgs(
48 opr, src, filter, opr->check_layout_fwd(src, filter, dst), bias, z,
49 dst) {}
50
51ConvBiasForwardImpl::AlgoBase::SizeArgs::SizeArgs(
52 ConvBiasForwardImpl* o, const TensorLayout& src, const TensorLayout& filter,

Callers

nothing calls this directly

Calls 1

check_layout_fwdMethod · 0.45

Tested by

no test coverage detected