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

Method check_exec_allow_noncontiguous

dnn/src/common/conv_bias.cpp:159–170  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

157}
158
159ConvBiasForward::CanonizedFilterMeta ConvBiasForward::check_exec_allow_noncontiguous(
160 const TensorLayout& src, const TensorLayout& filter, const TensorLayout& bias,
161 const TensorLayout& z, const TensorLayout& dst, size_t workspace_in_bytes,
162 const PreprocessedFilter* preprocessed_filter) {
163 do_check_exec_common(
164 this, src, filter, bias, z, dst, workspace_in_bytes, preprocessed_filter);
165 TensorLayout dst_expected;
166 dst_expected.dtype = dst.dtype;
167 auto ret = deduce_layout_fwd(src, filter, dst_expected);
168 megdnn_assert_eq_shape(dst_expected, dst);
169 return ret;
170}
171
172template <typename T>
173struct NCHWParamTrait;

Callers

nothing calls this directly

Calls 1

do_check_exec_commonFunction · 0.85

Tested by

no test coverage detected