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

Method check_layout

dnn/src/naive/convpooling/opr_impl.cpp:44–54  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

42}
43
44void ConvPoolingForwardImpl::check_layout(
45 const TensorLayout& src, const TensorLayout& filter, const TensorLayout& bias,
46 TensorLayout& dst, size_t /*workspace_limit_in_bytes*/) {
47 TensorLayout dst_expected;
48 this->deduce_layout(src, filter, bias, dst_expected);
49 megdnn_assert_eq_layout(dst_expected, dst);
50 megdnn_assert(bias.shape[1] == dst.shape[1]);
51 megdnn_assert(dst.shape[1] == filter.shape[0]);
52 // megdnn_assert_eq_layout(workspace_expected, workspace);
53 return;
54}
55
56void ConvPoolingForwardImpl::deduce_layout(
57 const TensorLayout& srcl, const TensorLayout& filterl,

Callers

nothing calls this directly

Calls 1

deduce_layoutMethod · 0.95

Tested by

no test coverage detected