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

Method check_layout

dnn/src/cuda/convpooling/opr_impl.cpp:88–98  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

86}
87
88void ConvPoolingForwardImpl::check_layout(
89 const TensorLayout& src, const TensorLayout& filter, const TensorLayout& bias,
90 TensorLayout& dst, size_t /* workspace_limit_in_bytes */
91) {
92 TensorLayout dst_expected;
93 deduce_layout(src, filter, bias, dst_expected);
94 megdnn_assert_eq_layout(dst_expected, dst);
95
96 megdnn_assert(bias.shape[1] == dst.shape[1]);
97 megdnn_assert(dst.shape[1] == filter.shape[0]);
98}
99
100void ConvPoolingForwardImpl::exec(
101 const _megdnn_in TensorND src, const _megdnn_in TensorND filter,

Callers

nothing calls this directly

Calls 1

deduce_layoutFunction · 0.85

Tested by

no test coverage detected