| /dports/math/tblis/tblis-1.2.0/src/internal/3t/indexed/ |
| H A D | mult.cxx | 144 factor, conj_A, data_A, group_AB.dense_stride[0], in contract_block()
145 group_AC.dense_stride[0], {}, in contract_block()
146 conj_B, data_B, group_AB.dense_stride[1], in contract_block()
147 group_BC.dense_stride[0], {}, in contract_block()
148 T(1), false, data_C, group_AC.dense_stride[1], in contract_block()
149 group_BC.dense_stride[1], {}); in contract_block()
254 factor, conj_A, data_A, group_AB.dense_stride[0], in mult_block()
255 group_AC.dense_stride[0], in mult_block()
257 conj_B, data_B, group_AB.dense_stride[1], in mult_block()
258 group_BC.dense_stride[0], in mult_block()
[all …]
|
| /dports/math/tblis/tblis-1.2.0/src/internal/1t/indexed/ |
| H A D | add.cxx | 94 factor, conj_A, data_A, group_A.dense_stride[0], in trace_block()
95 group_AB.dense_stride[0], in trace_block()
96 T(1), false, data_B, {}, group_AB.dense_stride[1]); in trace_block()
124 stride_vector dense_stride_B = group_AB.dense_stride[1] + group_B.dense_stride[0]; in replicate_block()
150 factor, conj_A, data_A, {}, group_AB.dense_stride[0], in replicate_block()
151 T(1), false, data_B, group_B.dense_stride[0], in replicate_block()
152 group_AB.dense_stride[1]); in replicate_block()
198 factor, conj_A, data_A, {}, group_AB.dense_stride[0], in transpose_block()
199 T(1), false, data_B, {}, group_AB.dense_stride[1]); in transpose_block()
|
| H A D | util.hpp | 95 group.dense_stride[I].push_back(A.dense_stride(idx_A[i])); in assign_dense_idx_helper()
122 group.mixed_stride[j].push_back(A.dense_stride(idx_A[i])); in assign_mixed_or_batch_idx_helper()
153 std::array<stride_vector,N> dense_stride; member
167 tblis::fold(dense_len, dense_idx, dense_stride[I]...); in fold()
173 return detail::sort_by_stride(dense_stride[I]...); in sort_by_stride()
|
| H A D | dot.cxx | 82 conj_A, data_A, group_AB.dense_stride[0], in dot_block()
83 conj_B, data_B, group_AB.dense_stride[1], in dot_block()
|
| /dports/math/onednn/oneDNN-2.5.1/src/gpu/jit/conv/ |
| H A D | gemm_schedule.cpp | 86 dim_t dense_stride = 1; in map_from_bmnk() local
88 b.stride = stride_t(dense_stride); in map_from_bmnk()
89 dense_stride *= b.block; in map_from_bmnk()
|
| H A D | tensor.cpp | 276 stride_t dense_stride = 1; in split_into_dense_tile() local
285 if (dense_stride != b.stride) return tensor_t(); in split_into_dense_tile()
286 dense_stride = b.block * b.stride; in split_into_dense_tile()
306 stride_t dense_stride = 1; in split_into_max_tile() local
315 if (dense_stride != b.stride) break; in split_into_max_tile()
316 dense_stride = b.block * b.stride; in split_into_max_tile()
|
| H A D | tensor.hpp | 582 stride_t dense_stride(1); in desc_str() local
596 } else if (b.stride != dense_stride) { in desc_str()
601 dense_stride = b.stride * b.block; in desc_str()
|
| /dports/math/tblis/tblis-1.2.0/src/internal/1t/indexed_dpd/ |
| H A D | util.hpp | 202 dim_vector sort_by_stride(const std::array<stride_vector,N>& dense_stride, in sort_by_stride()
205 return detail::sort_by_stride(dense_stride[I]...); in sort_by_stride()
241 std::array<stride_vector,N> dense_stride; in dpd_index_group() local
242 dense_total_lengths_and_strides(dense_len, dense_stride, in dpd_index_group()
260 dense_stride_sub[i] = stl_ext::select_from(dense_stride[i], in dpd_index_group()
274 if (dense_stride[i][reorder[j]] == 1) in dpd_index_group()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_uni_reorder_utils.cpp | 271 ptrdiff_t dense_stride = 1; in prb_init() local
279 dense_stride = dense_stride * imd.dims[old.id[d + 1]]; in prb_init()
280 last_stride = dense_stride; in prb_init()
|
| /dports/math/tblis/tblis-1.2.0/src/external/marray/include/ |
| H A D | indexed_varray_view.hpp | 123 using base::dense_stride;
|
| H A D | indexed_varray.hpp | 127 using base::dense_stride;
|
| H A D | indexed_varray_base.hpp | 570 stride_type dense_stride(unsigned dim) const in dense_stride() function in MArray::indexed_varray_base
|