Searched refs:data_p_off (Results 1 – 6 of 6) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
| H A D | ref_pooling.cpp | 173 auto data_p_off = get_offset(dst_d, mb, oc, od, oh, ow); in execute_forward() local
186 dst[data_p_off] = cpu::saturate_and_round<data_t>(res); in execute_forward()
191 auto data_p_off = get_offset(dst_d, mb, oc, od, oh, ow); in execute_forward() local
203 dst[data_p_off] = cpu::saturate_and_round<data_t>(res); in execute_forward()
|
| H A D | ref_eltwise.cpp | 106 auto data_p_off = DATA_OFF(data_d, n, c, d, h, w); in execute_forward_generic() local
108 alg_kind, src[data_p_off], alpha, beta); in execute_forward_generic()
117 dst[data_p_off] = cpu::saturate_and_round<data_t>(res); in execute_forward_generic()
|
| H A D | ref_resampling.cpp | 154 const dim_t data_p_off = get_offset(dst_d, mb, ch, od, oh, ow); in execute_forward() local
196 args.dst_val = dst[data_p_off]; in execute_forward()
199 store_fn(res, dst, data_p_off); in execute_forward()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
| H A D | ref_eltwise.cpp | 106 auto data_p_off = DATA_OFF(data_d, n, c, d, h, w); in execute_forward_generic() local
108 alg_kind, src[data_p_off], alpha, beta); in execute_forward_generic()
117 dst[data_p_off] = cpu::saturate_and_round<data_t>(res); in execute_forward_generic()
|
| H A D | ref_resampling.cpp | 154 const dim_t data_p_off = get_offset(dst_d, mb, ch, od, oh, ow); in execute_forward() local
196 args.dst_val = dst[data_p_off]; in execute_forward()
199 store_fn(res, dst, data_p_off); in execute_forward()
|
| H A D | ref_pooling.cpp | 182 auto data_p_off = get_offset(dst_d, mb, oc, od, oh, ow); in execute_forward() local
194 dst[data_p_off] = cpu::saturate_and_round<data_t>(res); in execute_forward()
|