Home
last modified time | relevance | path

Searched refs:pooling_desc (Results 1 – 6 of 6) sorted by relevance

/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/gxm/src/
H A DPoolingXSMM.cpp23 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 …]
/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/poolingdriver/
H A Dlayer_example_f32.c237 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 …]
H A Dlayer_example_bf16.c247 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 …]
/dports/math/libxsmm/libxsmm-1.16.3/src/
H A Dlibxsmm_dnn_pooling.c16 LIBXSMM_API libxsmm_dnn_pooling* libxsmm_dnn_create_pooling(libxsmm_dnn_pooling_desc pooling_desc, … in libxsmm_dnn_create_pooling() argument
23 …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()
/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/gxm/include/
H A DPoolingXSMM.hpp28 libxsmm_dnn_pooling_desc pooling_desc; member in PoolXSMM
/dports/math/libxsmm/libxsmm-1.16.3/include/
H A Dlibxsmm_dnn_pooling.h48 LIBXSMM_API libxsmm_dnn_pooling* libxsmm_dnn_create_pooling(libxsmm_dnn_pooling_desc pooling_desc, …