Searched refs:dims_idx (Results 1 – 8 of 8) sorted by relevance
28 dims_t dims_idx; in off2dims_idx() local29 dims_idx.reserve(dims.size()); in off2dims_idx()32 dims_idx.insert(dims_idx.begin(), off % dims[i]); in off2dims_idx()36 return dims_idx; in off2dims_idx()
36 dims_t dims_idx = off2dims_idx(l_dims, l_offset); in dump_point_values() local37 ss << dims_idx; in dump_point_values()
187 int dims_idx = 0; in str2desc() local190 d.sdims.resize(dims_idx + 1); in str2desc()198 ++dims_idx; in str2desc()
190 int get_broadcast_mask(const dims_t &dims_idx) const { in get_broadcast_mask()195 broadcast_mask += dims_dst[d] == dims_idx[d] ? (1 << d) : 0; in get_broadcast_mask()
49 dims_t dims_idx; in off2dims_idx() local50 dims_idx.reserve(dims.size()); in off2dims_idx()53 dims_idx.insert(dims_idx.begin(), off % dims[i]); in off2dims_idx()57 return dims_idx; in off2dims_idx()
394 dims_t dims_idx = off2dims_idx(l_dims, args.idx); in add_additional_fwd_lnorm_check() local395 int64_t c = dims_idx[prb->ndims - 1]; in add_additional_fwd_lnorm_check()
396 dims_t dims_idx = off2dims_idx(l_dims, args.idx); in add_additional_fwd_lnorm_check() local397 int64_t c = dims_idx[prb->ndims - 1]; in add_additional_fwd_lnorm_check()