Searched refs:pooling_desc (Results 1 – 6 of 6) sorted by relevance
23 pooling_desc.C = gp->nInput; in PoolXSMM()24 pooling_desc.H = gp->iHeight; in PoolXSMM()25 pooling_desc.W = gp->iWidth; in PoolXSMM()28 pooling_desc.R = gp->kh; in PoolXSMM()29 pooling_desc.S = gp->kw; in PoolXSMM()83 int imoff = pooling_desc.N * pooling_desc.C * ifhp * ifwp; in forwardPropagate()92 imoff = pooling_desc.N * pooling_desc.C * ofhp * ofwp; in forwardPropagate()101 imoff = pooling_desc.N * pooling_desc.C * ofhp * ofwp; in forwardPropagate()224 int imoff = pooling_desc.N * pooling_desc.C * ifhp * ifwp; in backPropagate()233 imoff = pooling_desc.N * pooling_desc.C * ofhp * ofwp; in backPropagate()[all …]
237 pooling_desc.N = nImg; in main()238 pooling_desc.C = nFm; in main()239 pooling_desc.H = ifh; in main()240 pooling_desc.W = ifw; in main()241 pooling_desc.u = stride_h; in main()242 pooling_desc.v = stride_w; in main()243 pooling_desc.R = kh; in main()244 pooling_desc.S = kw; in main()245 pooling_desc.pad_h = pad_h; in main()246 pooling_desc.pad_w = pad_w; in main()[all …]
247 pooling_desc.N = nImg; in main()248 pooling_desc.C = nFm; in main()249 pooling_desc.H = ifh; in main()250 pooling_desc.W = ifw; in main()251 pooling_desc.u = stride_h; in main()252 pooling_desc.v = stride_w; in main()253 pooling_desc.R = kh; in main()254 pooling_desc.S = kw; in main()255 pooling_desc.pad_h = pad_h; in main()256 pooling_desc.pad_w = pad_w; in main()[all …]
16 LIBXSMM_API libxsmm_dnn_pooling* libxsmm_dnn_create_pooling(libxsmm_dnn_pooling_desc pooling_desc, … in libxsmm_dnn_create_pooling() argument23 …if ( ((pooling_desc.datatype_in == LIBXSMM_DNN_DATATYPE_BF16) && (pooling_desc.datatype_out == LIB… in libxsmm_dnn_create_pooling()24 …((pooling_desc.datatype_in == LIBXSMM_DNN_DATATYPE_F32) && (pooling_desc.datatype_out == LIBXSMM_D… in libxsmm_dnn_create_pooling()32 handle->desc = pooling_desc; in libxsmm_dnn_create_pooling()
28 libxsmm_dnn_pooling_desc pooling_desc; member in PoolXSMM
48 LIBXSMM_API libxsmm_dnn_pooling* libxsmm_dnn_create_pooling(libxsmm_dnn_pooling_desc pooling_desc, …