Searched refs:idle_pos (Results 1 – 4 of 4) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
| H A D | ref_reduction.cpp | 133 dims_t idle_pos, reduce_pos; in execute_ref() local
134 utils::l_dims_by_l_offset(idle_pos, l_offset, dst_mdw.dims(), ndims); in execute_ref()
135 const dim_t dst_off = dst_mdw.off_v(idle_pos); in execute_ref()
136 const dim_t src_idle_off = src_mdw.off_v(idle_pos); in execute_ref()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
| H A D | ref_reduction.cpp | 133 dims_t idle_pos, reduce_pos; in execute_ref() local
134 utils::l_dims_by_l_offset(idle_pos, l_offset, dst_mdw.dims(), ndims); in execute_ref()
135 const dim_t dst_off = dst_mdw.off_v(idle_pos); in execute_ref()
136 const dim_t src_idle_off = src_mdw.off_v(idle_pos); in execute_ref()
|
| /dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/reduction/ |
| H A D | ref_reduction.cpp | 106 dims_t idle_pos = off2dims_idx(dst_dims, f); in compute_ref() local
107 const int64_t dst_off = md_off_v(dst.md_, idle_pos.data()); in compute_ref()
108 const int64_t src_idle_off = md_off_v(src.md_, idle_pos.data()); in compute_ref()
|
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/reduction/ |
| H A D | ref_reduction.cpp | 106 dims_t idle_pos = off2dims_idx(dst_dims, f); in compute_ref() local
107 const int64_t dst_off = md_off_v(dst.md_, idle_pos.data()); in compute_ref()
108 const int64_t src_idle_off = md_off_v(src.md_, idle_pos.data()); in compute_ref()
|