Searched refs:wei_3d_dims (Results 1 – 6 of 6) sorted by relevance
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/conv/ |
H A D | deconv.cpp | 80 dnnl_dims_t wei_3d_dims = {prb->g, prb->oc / prb->g, prb->ic / prb->g, in init_pd() local 83 ? &wei_3d_dims[!prb->has_groups] in init_pd()
|
H A D | conv_dw_fusion.cpp | 51 dnnl_dims_t wei_3d_dims = {prb->g, prb->oc / prb->g, prb->ic / prb->g, in init_pd() local 54 ? &wei_3d_dims[!prb->has_groups] in init_pd()
|
H A D | conv.cpp | 566 dnnl_dims_t wei_3d_dims = {prb->g, prb->oc / prb->g, prb->ic / prb->g, in init_pd_custom() local 569 ? &wei_3d_dims[!prb->has_groups] in init_pd_custom()
|
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/conv/ |
H A D | deconv.cpp | 77 dnnl_dims_t wei_3d_dims = {prb->g, prb->oc / prb->g, prb->ic / prb->g, in init_pd() local 80 ? &wei_3d_dims[!prb->has_groups] in init_pd()
|
H A D | conv_dw_fusion.cpp | 51 dnnl_dims_t wei_3d_dims = {prb->g, prb->oc / prb->g, prb->ic / prb->g, in init_pd() local 54 ? &wei_3d_dims[!prb->has_groups] in init_pd()
|
H A D | conv.cpp | 471 dnnl_dims_t wei_3d_dims = {prb->g, prb->oc / prb->g, prb->ic / prb->g, in init_pd() local 474 ? &wei_3d_dims[!prb->has_groups] in init_pd()
|