Home
last modified time | relevance | path

Searched refs:expect_dims (Results 1 – 2 of 2) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/
H A Drnn.cpp215 CHECK(expect_dims(r.src_layer_desc, {T, N, SLC}, false)); in check_dim_consistency()
216 CHECK(expect_dims(r.src_iter_desc, {L, D, N, SIC})); in check_dim_consistency()
217 CHECK(expect_dims(r.src_iter_c_desc, {L, D, N, DHC})); in check_dim_consistency()
220 CHECK(expect_dims(r.weights_peephole_desc, {L, D, 3, DHC})); in check_dim_consistency()
222 CHECK(expect_dims(r.bias_desc, {L, D, G + extra_bias, DHC})); in check_dim_consistency()
223 CHECK(expect_dims(r.dst_layer_desc, {T, N, DLC}, false)); in check_dim_consistency()
224 CHECK(expect_dims(r.dst_iter_desc, {L, D, N, DIC})); in check_dim_consistency()
225 CHECK(expect_dims(r.dst_iter_c_desc, {L, D, N, DHC})); in check_dim_consistency()
229 CHECK(expect_dims(r.diff_src_iter_desc, {L, D, N, SIC})); in check_dim_consistency()
231 CHECK(expect_dims( in check_dim_consistency()
[all …]
/dports/math/onednn/oneDNN-2.5.1/src/common/
H A Drnn.cpp216 CHECK(expect_dims(r.src_layer_desc, {T, N, SLC}, false)); in check_dim_consistency()
217 CHECK(expect_dims(r.src_iter_desc, {L, D, N, SIC})); in check_dim_consistency()
218 CHECK(expect_dims(r.src_iter_c_desc, {L, D, N, DHC})); in check_dim_consistency()
221 CHECK(expect_dims(r.weights_peephole_desc, {L, D, 3, DHC})); in check_dim_consistency()
223 CHECK(expect_dims(r.bias_desc, {L, D, G + extra_bias, DHC})); in check_dim_consistency()
224 CHECK(expect_dims(r.dst_layer_desc, {T, N, DLC}, false)); in check_dim_consistency()
225 CHECK(expect_dims(r.dst_iter_desc, {L, D, N, DIC})); in check_dim_consistency()
226 CHECK(expect_dims(r.dst_iter_c_desc, {L, D, N, DHC})); in check_dim_consistency()
230 CHECK(expect_dims(r.diff_src_iter_desc, {L, D, N, SIC})); in check_dim_consistency()
232 CHECK(expect_dims( in check_dim_consistency()
[all …]