Home
last modified time | relevance | path

Searched refs:tensor_load (Results 1 – 24 of 24) sorted by relevance

/external/llvm-project/mlir/test/Dialect/Standard/
Dcanonicalize.mlir3 // Test case: Basic folding of tensor_load(tensor_to_memref(t)) -> t
9 %1 = tensor_load %0 : memref<?xf32>
13 // Test case: Basic folding of tensor_to_memref(tensor_load(m)) -> m
18 %0 = tensor_load %arg0 : memref<?xf32>
26 // CHECK: %[[TENSOR:.*]] = tensor_load %[[MEMREF_ADDRSPACE2]] : memref<?xf32, 2>
30 %0 = tensor_load %arg0 : memref<?xf32, 2>
35 // Test case: Basic folding of dim(tensor_load(m)) -> dim(m).
43 %0 = tensor_load %arg0 : memref<?xf32>
Dfunc-bufferize.mlir5 // CHECK: %[[TENSOR:.*]] = tensor_load %[[ARG]] : memref<f32>
14 // CHECK: %[[T1:.*]] = tensor_load %[[ARG]] : memref<f32>
18 // CHECK: %[[T2:.*]] = tensor_load %[[BBARG]] : memref<f32>
38 // CHECK: %[[TENSOR:.*]] = tensor_load %[[ARG]] : memref<f32>
Dbufferize.mlir26 // CHECK: %[[RET:.*]] = tensor_load %[[MEMREF]] : memref<?xindex>
52 // CHECK: %[[RET:.*]] = tensor_load %[[MEMREF]] : memref<16x?xindex>
83 // CHECK: %[[RET:.*]] = tensor_load %[[RET_MEMREF]] : memref<f32>
94 // CHECK: %[[RET:.*]] = tensor_load %[[CASTED]]
105 // CHECK: %[[RET:.*]] = tensor_load %[[CASTED_MEMREF]] : memref<2xf32>
116 // CHECK: %[[RET:.*]] = tensor_load %[[CASTED_MEMREF]] : memref<*xf32>
131 // CHECK: %[[RET:.*]] = tensor_load %[[MEMREF]]
Dtensor-constant-bufferize.mlir10 // CHECK: %[[TENSOR:.*]] = tensor_load %[[MEMREF]]
Dops.mlir107 %1 = tensor_load %0 : memref<2xf32>
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/tests/passes/
Dfusion_op_remover.mlir8 %0 = tensor_load %input1 : memref<10xf32>
9 %1 = tensor_load %input2 : memref<10xf32>
12 %3 = tensor_load %input3 : memref<10xf32>
/external/llvm-project/mlir/test/Dialect/SCF/
Dbufferize.mlir14 // CHECK: %[[RESULT_TENSOR:.*]] = tensor_load %[[RESULT_MEMREF:.*]] : memref<?xf32>
32 // CHECK: %[[TENSOR_ITER:.*]] = tensor_load %[[ITER]] : memref<f32>
36 // CHECK: %[[VAL_8:.*]] = tensor_load %[[VAL_9:.*]] : memref<f32>
67 // CHECK: %[[TENSOR_ITER:.*]] = tensor_load %[[MEMREF_ITER]] : memref<f32>
72 // CHECK: %[[TENSOR:.*]] = tensor_load %[[RESULT:.*]] : memref<f32>
/external/llvm-project/mlir/test/Transforms/
Dfinalizing-bufferize.mlir7 %0 = tensor_load %arg0 : memref<f32>
24 %0 = tensor_load %arg0 : memref<f32>
/external/llvm-project/mlir/test/Dialect/Linalg/
Dbufferize.mlir6 // - tensor_to_memref / tensor_load materializations are properly inserted
23 // CHECK: %[[RESULT:.*]] = tensor_load %[[RESULT_MEMREF]] : memref<4xf32>
177 // CHECK-NEXT: %[[RT0:.*]] = tensor_load %[[A0]] : memref<2x3xf32>
185 // CHECK-NEXT: %[[RT1:.*]] = tensor_load %[[A1]] : memref<2x?xf32>
222 // CHECK-NEXT: %[[RT0:.*]] = tensor_load %[[M0_COPY]] : memref<?x?xf32>
236 // CHECK-NEXT: %[[RT1:.*]] = tensor_load %[[M1_COPY]] : memref<?x?xf32>
Dsparse_1d.mlir30 // CHECK: %[[VAL_10:.*]] = tensor_load %[[VAL_6]] : memref<32xf32>
56 // CHECK: %[[VAL_10:.*]] = tensor_load %[[VAL_6]] : memref<32xf32>
122 // CHECK: %[[VAL_29:.*]] = tensor_load %[[VAL_10]] : memref<32xf32>
157 // CHECK: %[[VAL_19:.*]] = tensor_load %[[VAL_7]] : memref<32xf32>
190 // CHECK: %[[VAL_15:.*]] = tensor_load %[[VAL_8]] : memref<32xf32>
233 // CHECK: %[[VAL_12:.*]] = tensor_load %[[VAL_7]] : memref<32xf32>
261 // CHECK: %[[VAL_12:.*]] = tensor_load %[[VAL_7]] : memref<32xf32>
333 // CHECK: %[[VAL_33:.*]] = tensor_load %[[VAL_11]] : memref<32xf32>
366 // CHECK: %[[VAL_17:.*]] = tensor_load %[[VAL_9]] : memref<32xf32>
438 // CHECK: %[[VAL_33:.*]] = tensor_load %[[VAL_11]] : memref<32xf32>
[all …]
Dsparse_3d.mlir40 // CHECK: %[[VAL_16:.*]] = tensor_load %[[VAL_9]] : memref<32x16x8xf32>
74 // CHECK: %[[VAL_16:.*]] = tensor_load %[[VAL_9]] : memref<32x16x8xf32>
155 // CHECK: %[[VAL_40:.*]] = tensor_load %[[VAL_13]] : memref<32x16x8xf32>
197 // CHECK: %[[VAL_24:.*]] = tensor_load %[[VAL_11]] : memref<32x16x8xf32>
282 // CHECK: %[[VAL_42:.*]] = tensor_load %[[VAL_13]] : memref<32x16x8xf32>
324 // CHECK: %[[VAL_24:.*]] = tensor_load %[[VAL_11]] : memref<32x16x8xf32>
435 // CHECK: %[[VAL_60:.*]] = tensor_load %[[VAL_15]] : memref<32x16x8xf32>
480 // CHECK: %[[VAL_27:.*]] = tensor_load %[[VAL_12]] : memref<32x16x8xf32>
570 // CHECK: %[[VAL_45:.*]] = tensor_load %[[VAL_13]] : memref<32x16x8xf32>
613 // CHECK: %[[VAL_25:.*]] = tensor_load %[[VAL_11]] : memref<32x16x8xf32>
[all …]
Dsparse_2d.mlir37 // CHECK: %[[VAL_14:.*]] = tensor_load %[[VAL_8]] : memref<32x16xf32>
68 // CHECK: %[[VAL_14:.*]] = tensor_load %[[VAL_8]] : memref<32x16xf32>
144 // CHECK: %[[VAL_36:.*]] = tensor_load %[[VAL_12]] : memref<32x16xf32>
181 // CHECK: %[[VAL_20:.*]] = tensor_load %[[VAL_10]] : memref<32x16xf32>
262 // CHECK: %[[VAL_39:.*]] = tensor_load %[[VAL_12]] : memref<32x16xf32>
300 // CHECK: %[[VAL_21:.*]] = tensor_load %[[VAL_10]] : memref<32x16xf32>
407 // CHECK: %[[VAL_57:.*]] = tensor_load %[[VAL_14]] : memref<32x16xf32>
448 // CHECK: %[[VAL_24:.*]] = tensor_load %[[VAL_11]] : memref<32x16xf32>
625 // CHECK: %[[VAL_115:.*]] = tensor_load %[[VAL_15]] : memref<32x16xf32>
719 // CHECK: %[[VAL_71:.*]] = tensor_load %[[VAL_15]] : memref<32x16xf32>
[all …]
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_fuse_linalg.cc98 if (auto tensor_load = dyn_cast<TensorLoadOp>(definingOp)) { in runOnFunction() local
99 auto alias = tensor_load.memref(); in runOnFunction()
/external/llvm-project/mlir/test/Dialect/Shape/
Dbufferize.mlir12 // CHECK: %[[TENSOR:.*]] = tensor_load %[[MEMREF:.*]] : memref<2xf16>
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dlhlo-fuse-linalg.mlir378 // Confirm that tiling information is passed through tensor_load, tensor.cast
392 %2 = tensor_load %1 : memref<32xf32>
405 // CHECK: tensor_load
416 // TILED: tensor_load
427 // PLOOP: tensor_load
Dlhlo_ops.mlir460 %0 = tensor_load %input1 : memref<10xf32>
461 %1 = tensor_load %input2 : memref<10xf32>
463 %3 = tensor_load %input3 : memref<10xf32>
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/tests/
Dtanh.mlir16 // CHECK-NOT: tensor_load
Dbufferize.mlir72 %result = tensor_load %arg : memref<?xf32>
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/hlo_to_lhlo_with_xla/
Dops.mlir704 // CHECK: %[[VAR0:.*]] = tensor_load %[[ARG0]] : memref<f32>
705 // CHECK: %[[VAR1:.*]] = tensor_load %[[ARG1]] : memref<f32>
724 // CHECK: %[[VAL0:.*]] = tensor_load %{{.*}} : memref<f32>
725 // CHECK: %[[VAL1:.*]] = tensor_load %{{.*}} : memref<f32>
726 // CHECK: %[[VAL2:.*]] = tensor_load %{{.*}} : memref<f32>
Dhlo_text_to_lhlo_no_opt.hlotxt8 // CHECK: tensor_load %arg0 {minor_to_major = dense<[0, 1]> : tensor<2xindex>}
/external/llvm-project/mlir/test/IR/
Dcore-ops.mlir886 // CHECK: %[[TENSOR:.*]] = tensor_load %[[MEMREF:.*]] : memref<4x4xi32>
887 %1 = tensor_load %0 : memref<4x4xi32>
895 // CHECK: %[[TENSOR:.*]] = tensor_load %[[MEMREF:.*]] : memref<*xi32>
896 %1 = tensor_load %0 : memref<*xi32>
/external/llvm-project/mlir/include/mlir/Transforms/
DPasses.td301 remaining `tensor_load` and `tensor_to_memref` operations.
304 exist in pairs, i.e., all uses of `tensor_load` operations are
/external/llvm-project/mlir/include/mlir/Dialect/StandardOps/IR/
DOps.td3505 def TensorLoadOp : Std_Op<"tensor_load",
3524 %12 = tensor_load %10 : memref<4x?xf32, #layout, memspace0>
3541 /// The result of a tensor_load is always a tensor.
3601 The opposite of this op is tensor_load. Together, these two ops are useful
3606 `tensor_to_memref(tensor_load(%memref)) -> %memref`, which is the property
/external/llvm-project/mlir/docs/
DLangRef.md98 %C = tensor_load %C_m : memref<100x50xf32>