Home
last modified time | relevance | path

Searched refs:dim (Results 1 – 25 of 580) sorted by relevance

12345678910>>...24

/llvm-project-15.0.7/llvm/test/MC/AMDGPU/
H A Dgfx10_asm_mimg.s3 image_load v[0:3], v0, s[0:7] dmask:0xf dim:SQ_RSRC_IMG_1D unorm
27 image_load v[0:1], v0, s[0:7] dmask:0x9 dim:1D
30 image_load v0, v0, s[0:7] dmask:0x1 dim:SQ_RSRC_IMG_1D dlc
36 image_load v0, v0, s[0:7] dmask:0x1 dim:SQ_RSRC_IMG_1D slc
46 image_load v0, v[2:3], s[0:7] dmask:0x1 dim:2D
49 image_load v0, v[2:4], s[0:7] dmask:0x1 dim:3D
52 image_load v0, v[2:4], s[0:7] dmask:0x1 dim:CUBE
55 image_load v0, v[2:3], s[0:7] dmask:0x1 dim:1D_ARRAY
58 image_load v0, v[2:4], s[0:7] dmask:0x1 dim:2D_ARRAY
61 image_load v0, v[2:4], s[0:7] dmask:0x1 dim:2D_MSAA
[all …]
H A Dgfx11_asm_mimg.s3 image_load v[0:3], v0, s[0:7] dmask:0xf dim:SQ_RSRC_IMG_1D unorm
27 image_load v[0:1], v0, s[0:7] dmask:0x9 dim:1D
30 image_load v0, v0, s[0:7] dmask:0x1 dim:SQ_RSRC_IMG_1D dlc
36 image_load v0, v0, s[0:7] dmask:0x1 dim:SQ_RSRC_IMG_1D slc
46 image_load v0, v[2:3], s[0:7] dmask:0x1 dim:2D
49 image_load v0, v[2:4], s[0:7] dmask:0x1 dim:3D
52 image_load v0, v[2:4], s[0:7] dmask:0x1 dim:CUBE
55 image_load v0, v[2:3], s[0:7] dmask:0x1 dim:1D_ARRAY
58 image_load v0, v[2:4], s[0:7] dmask:0x1 dim:2D_ARRAY
61 image_load v0, v[2:4], s[0:7] dmask:0x1 dim:2D_MSAA
[all …]
H A Dgfx11_asm_mimg_err.s3 image_sample_d v[64:66], [v32, v16, v8, v4, v2, v1], s[4:11], s[100:103] dmask:0x7 dim:SQ_RSRC_IMG_…
9 image_sample_d v[64:66], [v32, v16, v8, v4, v2, v1, v5], s[4:11], s[100:103] dmask:0x7 dim:SQ_RSRC_…
60 image_gather4_l_o v[64:67], [v32, v0, v4, v5, v6], s[4:11], s[100:103] dmask:0x1 dim:SQ_RSRC_IMG_3D
63 image_gather4_b_o v[64:67], [v32, v0, v4, v5, v6], s[4:11], s[100:103] dmask:0x1 dim:SQ_RSRC_IMG_3D
69 image_gather4_c_o v[64:67], [v32, v0, v4, v5, v6], s[4:11], s[100:103] dmask:0x1 dim:SQ_RSRC_IMG_3D
108 image_atomic_fcmpswap v[4:5], v32, s[96:103] dmask:0x3 dim:SQ_RSRC_IMG_1D glc
111 image_atomic_fmin v4, v32, s[96:103] dmask:0x1 dim:SQ_RSRC_IMG_1D glc
114 image_atomic_fmax v4, v32, s[96:103] dmask:0x1 dim:SQ_RSRC_IMG_1D glc
117 image_msaa_load v[1:4], v[5:7], s[8:15] dmask:0xf dim:SQ_RSRC_IMG_2D_MSAA
120 image_msaa_load v5, v[1:3], s[8:15] dmask:0x1 dim:SQ_RSRC_IMG_2D_MSAA d16
[all …]
H A Dgfx10_asm_mimg_err.s7 image_load v[0:3], v0, s[0:7] dmask:0xf dim:SQ_RSRC_IMG_1D da
10 image_load_pck v[0:3], v0, s[0:7] dmask:0xf dim:SQ_RSRC_IMG_1D d16
13 image_load v[0:1], v0, s[0:7] dmask:0xf dim:SQ_RSRC_IMG_1D
16 image_load v[0:3], v[0:1], s[0:7] dmask:0xf dim:SQ_RSRC_IMG_1D
19 image_load_mip v[0:3], v[0:2], s[0:7] dmask:0xf dim:SQ_RSRC_IMG_CUBE
22 image_sample_d v[0:3], [v0, v1, v2, v3, v4], s[0:7], s[8:11] dmask:0xf dim:SQ_RSRC_IMG_1D_ARRAY
25 image_sample_b_cl v[0:3], v[0:3], s[0:7], s[8:11] dmask:0xf dim:SQ_RSRC_IMG_CUBE
28 image_sample_c_d v[0:3], [v0, v1, v2, v3, v4, v5, v6], s[0:7], s[8:11] dmask:0xf dim:SQ_RSRC_IMG_2D…
31 image_sample_c_d_cl v[0:3], [v0, v1, v2, v3, v4, v5, v6, v7], s[0:7], s[8:11] dmask:0xf dim:SQ_RSRC…
34 image_sample_c_d_cl_o v[0:3], v[0:7], s[0:7], s[8:11] dmask:0xf dim:SQ_RSRC_IMG_2D
[all …]
H A Dgfx1013.s27 image_msaa_load v[5:6], v[1:4], s[8:15] dmask:0x3 dim:SQ_RSRC_IMG_2D_MSAA_ARRAY
30 image_msaa_load v[5:7], v[1:4], s[8:15] dmask:0x3 dim:SQ_RSRC_IMG_2D_MSAA_ARRAY tfe
33 image_msaa_load v5, v[1:4], s[8:15] dmask:0x8 dim:SQ_RSRC_IMG_2D_MSAA_ARRAY d16
36 image_msaa_load v[5:6], v[1:2], s[8:15] dmask:0x3 dim:SQ_RSRC_IMG_2D_MSAA_ARRAY a16
39 image_msaa_load v[5:6], v[1:3], s[8:15] dmask:0x3 dim:SQ_RSRC_IMG_2D_MSAA
42 image_msaa_load v[5:6], v[1:4], s[8:15] dmask:0x3 dim:SQ_RSRC_IMG_2D_MSAA_ARRAY unorm
45 image_msaa_load v[5:6], v[1:4], s[8:15] dmask:0x3 dim:SQ_RSRC_IMG_2D_MSAA_ARRAY glc
48 image_msaa_load v[5:6], v[1:4], s[8:15] dmask:0x3 dim:SQ_RSRC_IMG_2D_MSAA_ARRAY slc
51 image_msaa_load v[5:6], v[1:4], s[8:15] dmask:0x3 dim:SQ_RSRC_IMG_2D_MSAA_ARRAY dlc
54 image_msaa_load v[5:6], v[1:4], s[8:15] dmask:0x3 dim:SQ_RSRC_IMG_2D_MSAA_ARRAY lwe
/llvm-project-15.0.7/flang/include/flang/Runtime/
H A Dreduction.h81 const char *source, int line, int dim = 0,
84 const char *source, int line, int dim = 0,
87 const char *source, int line, int dim = 0,
90 const char *source, int line, int dim = 0,
93 const char *source, int line, int dim = 0,
96 const char *source, int line, int dim = 0,
114 const char *source, int line, int dim = 0,
137 const char *source, int line, int dim = 0,
140 const char *source, int line, int dim = 0,
143 const char *source, int line, int dim = 0,
[all …]
/llvm-project-15.0.7/flang/runtime/
H A Dreduction.cpp230 if (dim < 0 || dim > 1) { in GetTotalLogicalReduction()
249 const auto &dim{x.GetDimension(zeroBasedDim)}; in ReduceLogicalDimToScalar() local
250 SubscriptValue at{dim.LowerBound()}; in ReduceLogicalDimToScalar()
251 for (auto n{dim.Extent()}; n-- > 0; ++at) { in ReduceLogicalDimToScalar()
274 x, dim - 1, at); in operator ()()
329 return GetTotalLogicalReduction(x, source, line, dim, in RTNAME()
336 result, x, dim, terminator, "ALL"); in RTNAME()
340 return GetTotalLogicalReduction(x, source, line, dim, in RTNAME()
347 result, x, dim, terminator, "ANY"); in RTNAME()
360 kind, terminator, result, x, dim, terminator); in RTNAME()
[all …]
H A Dproduct.cpp58 const Descriptor &x, const char *source, int line, int dim, in RTNAME()
65 const Descriptor &x, const char *source, int line, int dim, in RTNAME()
72 const Descriptor &x, const char *source, int line, int dim, in RTNAME()
79 const Descriptor &x, const char *source, int line, int dim, in RTNAME()
87 const Descriptor &x, const char *source, int line, int dim, in RTNAME()
98 const char *source, int line, int dim, const Descriptor *mask) { in RTNAME()
126 const Descriptor &x, const char *source, int line, int dim, in RTNAME()
133 const Descriptor &x, const char *source, int line, int dim, in RTNAME()
141 const Descriptor &x, const char *source, int line, int dim, in RTNAME()
149 const Descriptor &x, const char *source, int line, int dim, in RTNAME()
[all …]
H A Dextrema.cpp361 if (dim == 0 || x.rank() == 1) { in DoMaxMinNorm2()
478 x, source, line, dim, mask, "MAXVAL"); in RTNAME()
483 x, source, line, dim, mask, "MAXVAL"); in RTNAME()
488 x, source, line, dim, mask, "MAXVAL"); in RTNAME()
493 x, source, line, dim, mask, "MAXVAL"); in RTNAME()
500 x, source, line, dim, mask, "MAXVAL"); in RTNAME()
508 x, source, line, dim, mask, "MAXVAL"); in RTNAME()
513 x, source, line, dim, mask, "MAXVAL"); in RTNAME()
519 x, source, line, dim, mask, "MAXVAL"); in RTNAME()
526 x, source, line, dim, mask, "MAXVAL"); in RTNAME()
[all …]
H A Dsum.cpp94 const char *source, int line, int dim, const Descriptor *mask) { in RTNAME()
99 const char *source, int line, int dim, const Descriptor *mask) { in RTNAME()
104 const char *source, int line, int dim, const Descriptor *mask) { in RTNAME()
109 const char *source, int line, int dim, const Descriptor *mask) { in RTNAME()
115 const char *source, int line, int dim, const Descriptor *mask) { in RTNAME()
124 const char *source, int line, int dim, const Descriptor *mask) { in RTNAME()
129 const char *source, int line, int dim, const Descriptor *mask) { in RTNAME()
149 const Descriptor &x, const char *source, int line, int dim, in RTNAME()
155 const Descriptor &x, const char *source, int line, int dim, in RTNAME()
162 const Descriptor &x, const char *source, int line, int dim, in RTNAME()
[all …]
H A DISO_Fortran_binding.cpp31 const CFI_dim_t *dim{descriptor->dim}; in CFI_address() local
33 p += (subscripts[j] - dim->lower_bound) * dim->sm; in CFI_address()
69 CFI_dim_t *dim{descriptor->dim}; in CFI_allocate() local
76 dim->extent = extent; in CFI_allocate()
77 dim->sm = byteSize; in CFI_allocate()
331 const CFI_dim_t &dim{source->dim[j]}; in CFI_section() local
368 result->dim[resRank].sm = actualStride[j] * source->dim[j].sm; in CFI_section()
403 result->dim[j].lower_bound = 0; in CFI_select_part()
404 result->dim[j].extent = source->dim[j].extent; in CFI_select_part()
405 result->dim[j].sm = source->dim[j].sm; in CFI_select_part()
[all …]
H A Dreduction-templates.h43 inline void DoTotalReduction(const Descriptor &x, int dim, in DoTotalReduction() argument
46 if (dim < 0 || dim > 1) { in DoTotalReduction()
48 intrinsic, dim, x.rank()); in DoTotalReduction()
79 const char *source, int line, int dim, const Descriptor *mask, in GetTotalReduction() argument
119 const auto &dim{x.GetDimension(zeroBasedDim)}; in ReduceDimToScalar()
120 SubscriptValue at{dim.LowerBound()}; in ReduceDimToScalar()
121 for (auto n{dim.Extent()}; n-- > 0; ++at) { in ReduceDimToScalar()
168 if (dim < 1 || dim > xRank) { in CreatePartialReductionResult()
172 int zeroBasedDim{dim - 1}; in CreatePartialReductionResult()
198 result, x, dim, terminator, intrinsic, TypeCode{CAT, KIND}); in PartialReduction()
[all …]
/llvm-project-15.0.7/llvm/test/Transforms/LoopVectorize/X86/
H A Dregister-assumption.ll8 br label %loop_exit.dim.11.critedge
10 loop_exit.dim.11.critedge: ; preds = %loop_body.dim.0
14 br label %loop_header.dim.017.preheader
16 loop_header.dim.017.preheader: ; preds = %loop_exit.dim.016, %loop_exit.dim.11.c…
17 br label %loop_body.dim.018
19 loop_body.dim.018: ; preds = %loop_body.dim.018, %loop_header.dim.01…
20 …%invar_address.dim.019.0135 = phi i64 [ 0, %loop_header.dim.017.preheader ], [ %0, %loop_body.dim.…
24 %0 = add nuw nsw i64 %invar_address.dim.019.0135, 1
26 br i1 %1, label %loop_header.dim.017.preheader, label %loop_body.dim.018
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaFoldConstantTranspose.cpp64 for (int dim = inputType.getRank() - 1; dim >= 0; --dim) { in matchAndRewrite() local
65 srcIndices[dim] = totalCount % inputShape[dim]; in matchAndRewrite()
66 totalCount /= inputShape[dim]; in matchAndRewrite()
70 for (int dim = outputType.getRank() - 1; dim >= 0; --dim) in matchAndRewrite() local
71 dstIndices[dim] = srcIndices[permValues[dim]]; in matchAndRewrite()
74 for (int dim = 1; dim < outputType.getRank(); ++dim) in matchAndRewrite() local
75 dstLinearIndex = dstLinearIndex * outputShape[dim] + dstIndices[dim]; in matchAndRewrite()
/llvm-project-15.0.7/llvm/unittests/Analysis/Inputs/ir2native_x86_64_model/
H A Dsaved_model.pbtxt165 dim {
168 dim {
190 dim {
193 dim {
229 dim {
251 dim {
287 dim {
290 dim {
351 dim {
658 dim {
[all …]
/llvm-project-15.0.7/polly/lib/External/isl/
H A Dbasis_reduction_tab.c28 unsigned dim; member
102 #define GBR_lp_set_obj(lp, obj, dim) set_lp_obj(lp, obj, dim) argument
107 #define GBR_lp_add_row(lp, row, dim) add_lp_row(lp, row, dim) argument
120 unsigned dim; in gbr_tab() local
126 dim = tab->n_var; in gbr_tab()
151 lp->dim = tab->n_var; in init_lp()
188 isl_seq_neg(lp->row->el + 1 + lp->dim, lp->obj, lp->dim); in solve_lp()
199 isl_seq_inner_product(lp->obj, sample->el + 1 + lp->dim, lp->dim, &lp->tmp2); in solve_lp()
226 isl_seq_clr(lp->row->el + 1 + lp->dim, lp->dim); in cut_lp_to_hyperplane()
236 isl_seq_cpy(lp->row->el + 1 + lp->dim, row, lp->dim); in cut_lp_to_hyperplane()
[all …]
H A Disl_convex_hull.c205 if (dim < 0) in wrap_constraints()
208 dim += 1; in wrap_constraints()
227 isl_seq_clr(lp->eq[k]+1+dim*i+2, dim-2); in wrap_constraints()
237 isl_seq_cpy(lp->eq[k]+1+dim*i, set->p[i]->eq[j], dim); in wrap_constraints()
238 isl_seq_clr(lp->eq[k]+1+dim*(i+1), dim*(set->n-i-1)); in wrap_constraints()
245 isl_seq_clr(lp->ineq[k]+1+dim*(i+1), dim*(set->n-i-1)); in wrap_constraints()
321 if (dim < 0) in isl_set_wrap_facet()
327 dim += 1; in isl_set_wrap_facet()
348 isl_seq_clr(obj->block.data + 1 + dim*i+3, dim-3); in isl_set_wrap_facet()
398 if (dim < 0) in initial_facet_constraint()
[all …]
H A Disl_local_space.c221 cmp = isl_space_cmp(ls1->dim, ls2->dim); in isl_local_space_cmp()
236 isl_size dim = isl_space_dim(ls->dim, isl_dim_all); in isl_local_space_dim() local
476 ls->dim = isl_space_set_tuple_id(ls->dim, type, id); in isl_local_space_set_tuple_id()
814 if (!isl_space_is_equal(ls1->dim, ls2->dim)) in isl_local_space_intersect()
906 ls->dim = isl_space_domain(ls->dim); in isl_local_space_domain()
925 ls->dim = isl_space_range(ls->dim); in isl_local_space_range()
940 ls->dim = isl_space_from_domain(ls->dim); in isl_local_space_from_domain()
1182 if (!isl_space_is_equal(ls->dim, subs->ls->dim)) in isl_local_space_substitute()
1623 ls->dim = isl_space_flatten_domain(ls->dim); in isl_local_space_flatten_domain()
1647 ls->dim = isl_space_flatten_range(ls->dim); in isl_local_space_flatten_range()
[all …]
/llvm-project-15.0.7/llvm/test/MC/Disassembler/AMDGPU/
H A Dgfx10_mimg.txt7 # GFX10: image_load v[0:3], v0, s[0:7] dmask:0xf dim:SQ_RSRC_IMG_1D unorm ; encoding: [0x00,0x1f,0x…
10 # GFX10: image_load v[0:3], v[1:2], s[0:7] dmask:0xf dim:SQ_RSRC_IMG_2D unorm ; encoding: [0x08,0x1…
13 # GFX10: image_load v[252:255], v[2:4], s[0:7] dmask:0xf dim:SQ_RSRC_IMG_3D unorm ; encoding: [0x10…
25 # GFX10: image_load v255, v[6:8], s[0:7] dmask:0x8 dim:SQ_RSRC_IMG_2D_MSAA unorm ; encoding: [0x30,…
50 # GFX10: image_load v16, v[8:9], s[96:103] dmask:0x1 dim:SQ_RSRC_IMG_2D ; encoding: [0x08,0x01,0x00…
53 # GFX10: image_load v[16:17], v[8:9], s[96:103] dmask:0x3 dim:SQ_RSRC_IMG_2D ; encoding: [0x08,0x03…
56 # GFX10: image_load v[16:18], v[8:9], s[96:103] dmask:0x7 dim:SQ_RSRC_IMG_2D ; encoding: [0x08,0x07…
59 # GFX10: image_load v[16:19], v[8:9], s[96:103] dmask:0xf dim:SQ_RSRC_IMG_2D ; encoding: [0x08,0x0f…
74 # GFX10: image_load v16, v[8:9], s[96:103] dmask:0x1 dim:SQ_RSRC_IMG_2D d16 ; encoding: [0x08,0x01,…
77 # GFX10: image_load v16, v[8:9], s[96:103] dmask:0x3 dim:SQ_RSRC_IMG_2D d16 ; encoding: [0x08,0x03,…
[all …]
H A Dmimg_gfx10.txt3 # GFX10: image_load v[4:6], v238, s[28:35] dmask:0x7 dim:SQ_RSRC_IMG_1D unorm ; encoding: [0x00,0x1…
6 # GFX10: image_load_pck v5, v0, s[8:15] dmask:0x1 dim:SQ_RSRC_IMG_1D glc ; encoding: [0x00,0x21,0x0…
9 # GFX10: image_load_pck_sgn v5, v0, s[8:15] dmask:0x1 dim:SQ_RSRC_IMG_1D lwe ; encoding: [0x00,0x01…
12 # GFX10: image_load_mip v5, v[0:1], s[8:15] dim:SQ_RSRC_IMG_1D ; encoding: [0x00,0x00,0x04,0xf0,0x0…
15 # GFX10: image_load_mip_pck v5, v[1:2], s[8:15] dmask:0x1 dim:SQ_RSRC_IMG_1D ; encoding: [0x00,0x01…
24 # GFX10: image_store_pck v1, v2, s[12:19] dmask:0x1 dim:SQ_RSRC_IMG_1D unorm ; encoding: [0x00,0x11…
27 # GFX10: image_store_mip v1, v[2:3], s[12:19] dim:SQ_RSRC_IMG_1D ; encoding: [0x00,0x00,0x24,0xf0,0…
72 # GFX10: image_get_resinfo v5, v1, s[8:15] dmask:0x1 dim:SQ_RSRC_IMG_1D ; encoding: [0x00,0x01,0x38…
87 # GFX10: image_load v1, v1, s[16:23] dmask:0x1 dim:SQ_RSRC_IMG_2D a16 ; encoding: [0x08,0x01,0x00,0…
93 # GFX10: image_load v1, v1, s[16:23] dmask:0x1 dim:SQ_RSRC_IMG_2D a16 lwe ; encoding: [0x08,0x01,0x…
[all …]
H A Dgfx11_mimg.txt3 # GFX11: image_load v[0:3], v0, s[0:7] dmask:0xf dim:SQ_RSRC_IMG_1D unorm ; encoding: [0x80,0x0f,0x…
27 # GFX11: image_load v[0:1], v0, s[0:7] dmask:0x9 dim:SQ_RSRC_IMG_1D ; encoding: [0x00,0x09,0x00,0xf…
30 # GFX11: image_load v0, v0, s[0:7] dmask:0x1 dim:SQ_RSRC_IMG_1D dlc ; encoding: [0x00,0x21,0x00,0xf…
33 # GFX11: image_load v255, v0, s[0:7] dmask:0x1 dim:SQ_RSRC_IMG_1D glc ; encoding: [0x00,0x41,0x00,0…
36 # GFX11: image_load v0, v0, s[0:7] dmask:0x1 dim:SQ_RSRC_IMG_1D slc ; encoding: [0x00,0x11,0x00,0xf…
39 # GFX11: image_load v0, v255, s[0:7] dmask:0x6 dim:SQ_RSRC_IMG_1D d16 ; encoding: [0x00,0x06,0x02,0…
42 # GFX11: image_load v0, v255, s[0:7] dmask:0x1 dim:SQ_RSRC_IMG_1D r128 ; encoding: [0x00,0x81,0x00,…
45 # GFX11: image_load v0, v[2:3], s[0:7] dmask:0x1 dim:SQ_RSRC_IMG_2D ; encoding: [0x04,0x01,0x00,0xf…
48 # GFX11: image_load v0, v[2:4], s[0:7] dmask:0x1 dim:SQ_RSRC_IMG_3D ; encoding: [0x08,0x01,0x00,0xf…
51 # GFX11: image_load v0, v[2:4], s[0:7] dmask:0x1 dim:SQ_RSRC_IMG_CUBE ; encoding: [0x0c,0x01,0x00,0…
[all …]
/llvm-project-15.0.7/mlir/test/Dialect/Linalg/
H A Dresolve-shaped-type-result-dims.mlir8 %1 = tensor.dim %0, %c2 : tensor<4x5x?xf32>
9 %2 = tensor.dim %0, %c0 : tensor<4x5x?xf32>
22 %1 = tensor.dim %0, %c2 : tensor<4x5x?xf32>
35 %1 = tensor.dim %0, %c0 : tensor<?x?xf32>
36 %2 = tensor.dim %0, %c1 : tensor<?x?xf32>
63 %3 = tensor.dim %0, %c0 : tensor<?x?xf32>
64 %4 = tensor.dim %0, %c1 : tensor<?x?xf32>
99 %2 = tensor.dim %1, %c1 : tensor<?x?xf32>
115 %1 = tensor.dim %0, %c0 : tensor<?x?xf32>
116 %2 = tensor.dim %0, %c1 : tensor<?x?xf32>
[all …]
/llvm-project-15.0.7/polly/lib/Support/
H A DISLTools.cpp59 unsigned Dims1 = unsignedFromIslSize(FromSpace1.dim(isl::dim::set)); in makeTupleSwapBasicMap()
60 unsigned Dims2 = unsignedFromIslSize(FromSpace2.dim(isl::dim::set)); in makeTupleSwapBasicMap()
69 Result = Result.equate(isl::dim::in, i, isl::dim::out, Dims2 + i); in makeTupleSwapBasicMap()
71 Result = Result.equate(isl::dim::in, Dims1 + i, isl::dim::out, i); in makeTupleSwapBasicMap()
244 case isl::dim::in: in shiftDim()
474 Translator = Translator.equate(isl::dim::in, i, isl::dim::out, in distributeDomain()
614 unsigned ALen = unsignedFromIslSize(A.dim(isl::dim::set)); in flatCompare()
615 unsigned BLen = unsignedFromIslSize(B.dim(isl::dim::set)); in flatCompare()
621 unsignedFromIslSize(A.dim(isl::dim::param))) in flatCompare()
626 unsignedFromIslSize(B.dim(isl::dim::param))) in flatCompare()
[all …]
/llvm-project-15.0.7/mlir/test/Interfaces/InferShapedTypeOpInterface/
H A Dresolve-shaped-type-result-dims.mlir10 %1 = tensor.dim %0#0, %c0 : tensor<?x5xf32>
11 %2 = tensor.dim %0#0, %c1 : tensor<?x5xf32>
12 %3 = tensor.dim %0#1, %c0 : tensor<2x3x?xf32>
13 %4 = tensor.dim %0#1, %c1 : tensor<2x3x?xf32>
14 %5 = tensor.dim %0#1, %c2 : tensor<2x3x?xf32>
24 // CHECK-DAG: %[[D0:.+]] = tensor.dim %[[ARG_1]], %[[C0]]
37 %1 = tensor.dim %0#0, %c0 : tensor<?x5xf32>
38 %2 = tensor.dim %0#0, %c1 : tensor<?x5xf32>
39 %3 = tensor.dim %0#1, %c0 : tensor<2x3x?xf32>
40 %4 = tensor.dim %0#1, %c1 : tensor<2x3x?xf32>
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DConstantFold.cpp185 for (int dim = loopBounds.size() - 1; dim >= 0; --dim) { in matchAndRewrite() local
186 indices[dim] = totalCount % loopBounds[dim]; in matchAndRewrite()
187 totalCount /= loopBounds[dim]; in matchAndRewrite()
190 for (int dim = loopBounds.size() - 1; dim >= 0; --dim) { in matchAndRewrite() local
192 srcIndices[i][dim] = indices[inputDims[i][dim]]; in matchAndRewrite()
193 dstIndices[dim] = indices[outputDims[dim]]; in matchAndRewrite()
200 for (int dim = 1; dim < outputType.getRank(); ++dim) { in matchAndRewrite() local
201 dstLinearIndex = dstLinearIndex * outputShape[dim] + dstIndices[dim]; in matchAndRewrite()
204 srcLinearIndices[i] * inputShapes[i][dim] + srcIndices[i][dim]; in matchAndRewrite()

12345678910>>...24