/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
D | case.mlir | 3 func @main() -> tensor<f32> { 5 %cst_0 = constant dense<5.600000e+01> : tensor<f32> 6 %cst_1 = constant dense<1.200000e+01> : tensor<f32> 7 %cst_2 = constant dense<1.300000e+01> : tensor<f32> 9 ^bb0(%arg0: tensor<f32>): 10 %1 = "mhlo.negate"(%arg0) : (tensor<f32>) -> tensor<f32> 11 "mhlo.return"(%1) : (tensor<f32>) -> () 13 ^bb0(%arg0: tensor<f32>): 14 %1 = "mhlo.copy"(%arg0) : (tensor<f32>) -> tensor<f32> 15 "mhlo.return"(%1) : (tensor<f32>) -> () [all …]
|
D | if.mlir | 3 // CHECK: [[R0:%.+]] ([[A0:.+]]: (f32[])) -> (f32[]) { 4 // CHECK: %[[A0]] = (f32[]) parameter(0) 5 func @then_branch(%arg0: tuple<tensor<f32>>) -> tuple<tensor<f32>> { 6 // CHECK: %[[VAL0:.+]] = f32[] get-tuple-element((f32[]) %[[A0]]), index=0 7 %0 = "mhlo.get_tuple_element"(%arg0) {index = 0 : i32} : (tuple<tensor<f32>>) -> tensor<f32> 9 // CHECK: %[[VAL1:.+]] = f32[] log(f32[] %[[VAL0]]) 10 %1 = "mhlo.log"(%0) : (tensor<f32>) -> tensor<f32> 12 // CHECK: ROOT %[[VAl2:.+]] = (f32[]) tuple(f32[] %[[VAL1]]) 13 %2 = "mhlo.tuple"(%1) : (tensor<f32>) -> tuple<tensor<f32>> 14 return %2 : tuple<tensor<f32>> [all …]
|
D | fusion.hlotxt | 5 // CHECK: func @main(%[[ARG0:.*]]: tensor<f32>, %[[ARG1:.*]]: tensor<f32>) -> tensor<f32> { 7 // CHECK: ^bb0(%[[ARG2:.*]]: tensor<f32>, %[[ARG3:.*]]: tensor<f32>): 8 // CHECK: }) {fusion_kind = "kLoop"} : (tensor<f32>, tensor<f32>) -> tensor<f32> 10 // CHECK: ^bb0(%[[ARG2:.*]]: tensor<f32>, %[[ARG3:.*]]: tensor<f32>): 11 // CHECK: }) {fusion_kind = "kLoop"} : (tensor<f32>, tensor<f32>) -> tuple<tensor<f32>, tensor<f3… 14 %region_0.3 (Arg_0.4: f32[], Arg_1.5: f32[]) -> f32[] { 15 %Arg_0.4 = f32[] parameter(0) 16 %Arg_1.5 = f32[] parameter(1) 17 ROOT %add.6 = f32[] add(f32[] %Arg_0.4, f32[] %Arg_1.5) 20 %region_1.8 (Arg_0.9: f32[], Arg_1.10: f32[]) -> (f32[], f32[]) { [all …]
|
D | fusion.mlir | 3 // CHECK: %[[REGION0:.*]] ({{.*}}: f32[], {{.*}}: f32[]) -> f32[] 4 // CHECK: %[[REGION1:.*]] ({{.*}}: f32[], {{.*}}: f32[]) -> (f32[], f32[]) 7 // CHECK: %[[PARAM0:.*]] = f32[] parameter(0) 8 // CHECK: %[[PARAM1:.*]] = f32[] parameter(1) 9 // CHECK: %[[FUSION0:.*]] = f32[] fusion(f32[] %[[PARAM0]], f32[] %[[PARAM1]]), kind=kLoop, calls… 10 // CHECK: %[[FUSION1:.*]] = (f32[], f32[]) fusion(f32[] %[[PARAM0]], f32[] %[[PARAM1]]), kind=kLo… 11 // CHECK: f32[] get-tuple-element((f32[], f32[]) %[[FUSION1]]), index=0 12 // CHECK: f32[] get-tuple-element((f32[], f32[]) %[[FUSION1]]), index=1 14 func @main(%arg0: tensor<f32>, %arg1: tensor<f32>) { 16 ^bb0(%arg2: tensor<f32>, %arg3: tensor<f32>): [all …]
|
D | import.hlotxt | 8 // CHECK-LABEL: func @main(%arg0: tensor<f32>) -> tensor<f32> { 9 ENTRY %dummy_main (Arg_0.1: f32[]) -> f32[] { 10 ROOT %Arg_0.1 = f32[] parameter(0) 14 %test_simple (Arg_0.1: f32[4], Arg_1.2: f32[4]) -> f32[] { 15 %Arg_0.1 = f32[4]{0} parameter(0) 16 %Arg_1.2 = f32[4]{0} parameter(1) 19 %add.42 = f32[4]{0} add(f32[4]{0} %Arg_0.1, f32[4]{0} %Arg_1.2) 22 … %arg1) {precision_config = ["DEFAULT", "DEFAULT"]} : (tensor<4xf32>, tensor<4xf32>) -> tensor<f32> 23 …ROOT %dot.4 = f32[] dot(f32[4]{0} %add.42, f32[4]{0} %Arg_1.2), lhs_contracting_dims={0}, rhs_cont… 37 lhs = f32[] parameter(0) [all …]
|
D | case_conditional.hlotxt | 5 %Negate (x: f32[]) -> f32[] { 6 %x = f32[] parameter(0) 7 ROOT %negate = f32[] negate(f32[] %x) 10 %Identity (y: f32[]) -> f32[] { 11 %y = f32[] parameter(0) 12 ROOT %copy = f32[] copy(f32[] %y) 15 %Floor (z: f32[]) -> f32[] { 16 %z = f32[] parameter(0) 17 ROOT %floor = f32[] floor(f32[] %z) 20 ENTRY %indexed_conditional () -> f32[] { [all …]
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_parser_test.cc | 129 ENTRY %ConstantF32.v4 () -> f32[] { in CreateTestCases() 130 ROOT %constant = f32[] constant(42), backend_config="this is a configuration" in CreateTestCases() 140 ENTRY %ConstantF32Empty.v4 () -> f32[0] { in CreateTestCases() 141 ROOT %constant = f32[0]{0} constant({}) in CreateTestCases() 151 ENTRY %ConstantF32R4Empty.v4 () -> f32[2,0,4,3] { in CreateTestCases() 152 ROOT %constant = f32[2,0,4,3]{3,2,1,0} constant({ { /*i0=0*/ }, { /*i0=1*/ } }) in CreateTestCases() 162 ENTRY %Small_3x2x1x1.v1 () -> f32[3,2,1,1] { in CreateTestCases() 163 …ROOT %constant = f32[3,2,1,1]{3,2,1,0} constant({ { /*i0=0*/ { /*i1=0*/ {-1} }, { /*i1=1*/ {4.1} }… in CreateTestCases() 174 %constant = f32[6]{0} constant({nan, 7, nan, -1, inf, -inf}) in CreateTestCases() 175 ROOT %is-finite = pred[6]{0} is-finite(f32[6]{0} %constant) in CreateTestCases() [all …]
|
/external/rust/crates/libm/src/math/ |
D | erff.rs | 18 const ERX: f32 = 8.4506291151e-01; /* 0x3f58560b */ 22 const EFX8: f32 = 1.0270333290e+00; /* 0x3f8375d4 */ 23 const PP0: f32 = 1.2837916613e-01; /* 0x3e0375d4 */ 24 const PP1: f32 = -3.2504209876e-01; /* 0xbea66beb */ 25 const PP2: f32 = -2.8481749818e-02; /* 0xbce9528f */ 26 const PP3: f32 = -5.7702702470e-03; /* 0xbbbd1489 */ 27 const PP4: f32 = -2.3763017452e-05; /* 0xb7c756b1 */ 28 const QQ1: f32 = 3.9791721106e-01; /* 0x3ecbbbce */ 29 const QQ2: f32 = 6.5022252500e-02; /* 0x3d852a63 */ 30 const QQ3: f32 = 5.0813062117e-03; /* 0x3ba68116 */ [all …]
|
D | j0f.rs | 18 const INVSQRTPI: f32 = 5.6418961287e-01; /* 0x3f106ebb */ 19 const TPI: f32 = 6.3661974669e-01; /* 0x3f22f983 */ 21 fn common(ix: u32, x: f32, y0: bool) -> f32 { in common() argument 22 let z: f32; in common() 23 let s: f32; in common() 24 let mut c: f32; in common() 25 let mut ss: f32; in common() 26 let mut cc: f32; in common() 56 const R02: f32 = 1.5625000000e-02; /* 0x3c800000 */ 57 const R03: f32 = -1.8997929874e-04; /* 0xb947352e */ [all …]
|
D | lgammaf_r.rs | 18 const PI: f32 = 3.1415927410e+00; /* 0x40490fdb */ 19 const A0: f32 = 7.7215664089e-02; /* 0x3d9e233f */ 20 const A1: f32 = 3.2246702909e-01; /* 0x3ea51a66 */ 21 const A2: f32 = 6.7352302372e-02; /* 0x3d89f001 */ 22 const A3: f32 = 2.0580807701e-02; /* 0x3ca89915 */ 23 const A4: f32 = 7.3855509982e-03; /* 0x3bf2027e */ 24 const A5: f32 = 2.8905137442e-03; /* 0x3b3d6ec6 */ 25 const A6: f32 = 1.1927076848e-03; /* 0x3a9c54a1 */ 26 const A7: f32 = 5.1006977446e-04; /* 0x3a05b634 */ 27 const A8: f32 = 2.2086278477e-04; /* 0x39679767 */ [all …]
|
/external/llvm-project/llvm/test/MC/ARM/ |
D | thumb-neon-v8.s | 3 vmaxnm.f32 d4, d5, d1 4 @ CHECK: vmaxnm.f32 d4, d5, d1 @ encoding: [0x05,0xff,0x11,0x4f] 5 vmaxnm.f32 q2, q4, q6 6 @ CHECK: vmaxnm.f32 q2, q4, q6 @ encoding: [0x08,0xff,0x5c,0x4f] 7 vminnm.f32 d5, d4, d30 8 @ CHECK: vminnm.f32 d5, d4, d30 @ encoding: [0x24,0xff,0x3e,0x5f] 9 vminnm.f32 q0, q13, q2 10 @ CHECK: vminnm.f32 q0, q13, q2 @ encoding: [0x2a,0xff,0xd4,0x0f] 12 vcvta.s32.f32 d4, d6 13 @ CHECK: vcvta.s32.f32 d4, d6 @ encoding: [0xbb,0xff,0x06,0x40] [all …]
|
D | neon-v8.s | 3 vmaxnm.f32 d4, d5, d1 4 @ CHECK: vmaxnm.f32 d4, d5, d1 @ encoding: [0x11,0x4f,0x05,0xf3] 5 vmaxnm.f32 q2, q4, q6 6 @ CHECK: vmaxnm.f32 q2, q4, q6 @ encoding: [0x5c,0x4f,0x08,0xf3] 7 vminnm.f32 d5, d4, d30 8 @ CHECK: vminnm.f32 d5, d4, d30 @ encoding: [0x3e,0x5f,0x24,0xf3] 9 vminnm.f32 q0, q13, q2 10 @ CHECK: vminnm.f32 q0, q13, q2 @ encoding: [0xd4,0x0f,0x2a,0xf3] 12 vcvta.s32.f32 d4, d6 13 @ CHECK: vcvta.s32.f32 d4, d6 @ encoding: [0x06,0x40,0xbb,0xf3] [all …]
|
D | neon-convert-encoding.s | 3 @ CHECK: vcvt.s32.f32 d16, d16 @ encoding: [0x20,0x07,0xfb,0xf3] 4 vcvt.s32.f32 d16, d16 5 @ CHECK: vcvt.u32.f32 d16, d16 @ encoding: [0xa0,0x07,0xfb,0xf3] 6 vcvt.u32.f32 d16, d16 7 @ CHECK: vcvt.f32.s32 d16, d16 @ encoding: [0x20,0x06,0xfb,0xf3] 8 vcvt.f32.s32 d16, d16 9 @ CHECK: vcvt.f32.u32 d16, d16 @ encoding: [0xa0,0x06,0xfb,0xf3] 10 vcvt.f32.u32 d16, d16 11 @ CHECK: vcvt.s32.f32 q8, q8 @ encoding: [0x60,0x07,0xfb,0xf3] 12 vcvt.s32.f32 q8, q8 [all …]
|
/external/llvm/test/MC/ARM/ |
D | thumb-neon-v8.s | 3 vmaxnm.f32 d4, d5, d1 4 @ CHECK: vmaxnm.f32 d4, d5, d1 @ encoding: [0x05,0xff,0x11,0x4f] 5 vmaxnm.f32 q2, q4, q6 6 @ CHECK: vmaxnm.f32 q2, q4, q6 @ encoding: [0x08,0xff,0x5c,0x4f] 7 vminnm.f32 d5, d4, d30 8 @ CHECK: vminnm.f32 d5, d4, d30 @ encoding: [0x24,0xff,0x3e,0x5f] 9 vminnm.f32 q0, q13, q2 10 @ CHECK: vminnm.f32 q0, q13, q2 @ encoding: [0x2a,0xff,0xd4,0x0f] 12 vcvta.s32.f32 d4, d6 13 @ CHECK: vcvta.s32.f32 d4, d6 @ encoding: [0xbb,0xff,0x06,0x40] [all …]
|
D | neon-v8.s | 3 vmaxnm.f32 d4, d5, d1 4 @ CHECK: vmaxnm.f32 d4, d5, d1 @ encoding: [0x11,0x4f,0x05,0xf3] 5 vmaxnm.f32 q2, q4, q6 6 @ CHECK: vmaxnm.f32 q2, q4, q6 @ encoding: [0x5c,0x4f,0x08,0xf3] 7 vminnm.f32 d5, d4, d30 8 @ CHECK: vminnm.f32 d5, d4, d30 @ encoding: [0x3e,0x5f,0x24,0xf3] 9 vminnm.f32 q0, q13, q2 10 @ CHECK: vminnm.f32 q0, q13, q2 @ encoding: [0xd4,0x0f,0x2a,0xf3] 12 vcvta.s32.f32 d4, d6 13 @ CHECK: vcvta.s32.f32 d4, d6 @ encoding: [0x06,0x40,0xbb,0xf3] [all …]
|
D | neon-convert-encoding.s | 3 @ CHECK: vcvt.s32.f32 d16, d16 @ encoding: [0x20,0x07,0xfb,0xf3] 4 vcvt.s32.f32 d16, d16 5 @ CHECK: vcvt.u32.f32 d16, d16 @ encoding: [0xa0,0x07,0xfb,0xf3] 6 vcvt.u32.f32 d16, d16 7 @ CHECK: vcvt.f32.s32 d16, d16 @ encoding: [0x20,0x06,0xfb,0xf3] 8 vcvt.f32.s32 d16, d16 9 @ CHECK: vcvt.f32.u32 d16, d16 @ encoding: [0xa0,0x06,0xfb,0xf3] 10 vcvt.f32.u32 d16, d16 11 @ CHECK: vcvt.s32.f32 q8, q8 @ encoding: [0x60,0x07,0xfb,0xf3] 12 vcvt.s32.f32 q8, q8 [all …]
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
D | gemm_rewrite_test.cc | 83 x = f32[2,2] parameter(0) 84 y = f32[2,2] parameter(1) 85 x_transposed = f32[2,2] transpose(x), dimensions={1, 0} 86 ROOT dot_a = f32[2,2] dot(x_transposed, y), lhs_contracting_dims={1}, rhs_contracting_dims={0} 95 ; CHECK-LABEL: ENTRY %AddDotsFunc (x: f32[2,2], y: f32[2,2]) -> f32[2,2] { 96 ; CHECK-NEXT: %x = f32[2,2]{1,0} parameter(0) 97 ; CHECK-NEXT: %y = f32[2,2]{1,0} parameter(1) 98 ; CHECK-NEXT: ROOT %custom-call = f32[2,2]{1,0} custom-call(%x, %y), custom_call_target="__cubla… 107 x = f32[2,2] parameter(0) 108 y = f32[2,2] parameter(1) [all …]
|
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/ |
D | legalize-trigonometric-to-approximation.mlir | 13 func @tanh_f32(%arg0 : f32) -> f32 { 14 %res = math.tanh %arg0 : f32 15 return %res : f32 21 // CHECK-SAME: (%[[VAL_0:.*]]: f32) -> f32 22 // CHECK: %[[VAL_1:.*]] = constant 4.000000e-04 : f32 23 // CHECK: %[[VAL_2:.*]] = constant 7.90531111 : f32 24 // CHECK: %[[VAL_3:.*]] = constant -7.90531111 : f32 25 // CHECK: %[[VAL_4:.*]] = constant -2.76076837E-16 : f32 26 // CHECK: %[[VAL_5:.*]] = constant 2.00018794E-13 : f32 27 // CHECK: %[[VAL_6:.*]] = constant -8.60467184E-11 : f32 [all …]
|
/external/XNNPACK/scripts/ |
D | generate-f32-vrnd.sh | 9 tools/xngen src/f32-vrnd/scalar-libm.c.in -D OP=RNDNE -D BATCH_TILE=1 -o src/f32-vrnd/gen/vrndne-sc… 10 tools/xngen src/f32-vrnd/scalar-libm.c.in -D OP=RNDNE -D BATCH_TILE=2 -o src/f32-vrnd/gen/vrndne-sc… 11 tools/xngen src/f32-vrnd/scalar-libm.c.in -D OP=RNDNE -D BATCH_TILE=4 -o src/f32-vrnd/gen/vrndne-sc… 12 tools/xngen src/f32-vrnd/scalar-libm.c.in -D OP=RNDZ -D BATCH_TILE=1 -o src/f32-vrnd/gen/vrndz-sca… 13 tools/xngen src/f32-vrnd/scalar-libm.c.in -D OP=RNDZ -D BATCH_TILE=2 -o src/f32-vrnd/gen/vrndz-sca… 14 tools/xngen src/f32-vrnd/scalar-libm.c.in -D OP=RNDZ -D BATCH_TILE=4 -o src/f32-vrnd/gen/vrndz-sca… 15 tools/xngen src/f32-vrnd/scalar-libm.c.in -D OP=RNDU -D BATCH_TILE=1 -o src/f32-vrnd/gen/vrndu-sca… 16 tools/xngen src/f32-vrnd/scalar-libm.c.in -D OP=RNDU -D BATCH_TILE=2 -o src/f32-vrnd/gen/vrndu-sca… 17 tools/xngen src/f32-vrnd/scalar-libm.c.in -D OP=RNDU -D BATCH_TILE=4 -o src/f32-vrnd/gen/vrndu-sca… 18 tools/xngen src/f32-vrnd/scalar-libm.c.in -D OP=RNDD -D BATCH_TILE=1 -o src/f32-vrnd/gen/vrndd-sca… [all …]
|
D | generate-f32-vbinary.sh | 9 tools/xngen src/f32-vbinary/vop-scalar.c.in -D OP=ADD -D BATCH_TILE=1 -D WASM=0 -D ACTIVATION=M… 10 tools/xngen src/f32-vbinary/vop-scalar.c.in -D OP=ADD -D BATCH_TILE=2 -D WASM=0 -D ACTIVATION=M… 11 tools/xngen src/f32-vbinary/vop-scalar.c.in -D OP=ADD -D BATCH_TILE=4 -D WASM=0 -D ACTIVATION=M… 12 tools/xngen src/f32-vbinary/vop-scalar.c.in -D OP=ADD -D BATCH_TILE=8 -D WASM=0 -D ACTIVATION=M… 13 tools/xngen src/f32-vbinary/vop-scalar.c.in -D OP=DIV -D BATCH_TILE=1 -D WASM=0 -D ACTIVATION=M… 14 tools/xngen src/f32-vbinary/vop-scalar.c.in -D OP=DIV -D BATCH_TILE=2 -D WASM=0 -D ACTIVATION=M… 15 tools/xngen src/f32-vbinary/vop-scalar.c.in -D OP=DIV -D BATCH_TILE=4 -D WASM=0 -D ACTIVATION=M… 16 tools/xngen src/f32-vbinary/vop-scalar.c.in -D OP=DIV -D BATCH_TILE=8 -D WASM=0 -D ACTIVATION=M… 17 tools/xngen src/f32-vbinary/vop-scalar.c.in -D OP=MUL -D BATCH_TILE=1 -D WASM=0 -D ACTIVATION=M… 18 tools/xngen src/f32-vbinary/vop-scalar.c.in -D OP=MUL -D BATCH_TILE=2 -D WASM=0 -D ACTIVATION=M… [all …]
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
D | promote_resources_to_args.mlir | 4 // CHECK-LABEL: func @main(%arg0: tensor<i1>, %arg1: tensor<f32> {tf.resource_name = "x"}) -> tenso… 11 %0 = "tf.Const"() {value = dense<4.200000e+01> : tensor<f32>} : () -> tensor<f32> 12 … = "tf.VarHandleOp"() {container = "", shared_name = "x"} : () -> tensor<!tf.resource<tensor<f32>>> 13 %2 = "tf.ReadVariableOp"(%1) : (tensor<!tf.resource<tensor<f32>>>) -> tensor<f32> 14 %3 = "tf.AddV2"(%2, %0) : (tensor<f32>, tensor<f32>) -> tensor<f32> 15 %4 = "tf.Pack"(%0, %3) : (tensor<f32>, tensor<f32>) -> tensor<2xf32> 22 // CHECK-LABEL: func @main(%arg0: tensor<i1>) -> (tensor<f32> {tf.resource_name = "x"}) 27 %0 = "tf.Const"() {value = dense<4.200000e+01> : tensor<f32>} : () -> tensor<f32> 28 …) {container = "", shape = "tfshape$", shared_name = "x"} : () -> tensor<!tf.resource<tensor<f32>>> 29 "tf.AssignVariableOp"(%1, %0) : (tensor<!tf.resource<tensor<f32>>>, tensor<f32>) -> () [all …]
|
/external/llvm-project/mlir/test/Dialect/Standard/ |
D | func-bufferize.mlir | 4 // CHECK-SAME: %[[ARG:.*]]: memref<f32>) -> memref<f32> { 5 // CHECK: %[[TENSOR:.*]] = tensor_load %[[ARG]] : memref<f32> 6 // CHECK: %[[MEMREF:.*]] = tensor_to_memref %[[TENSOR]] : memref<f32> 7 // CHECK: return %[[MEMREF]] : memref<f32> 8 func @identity(%arg0: tensor<f32>) -> tensor<f32> { 9 return %arg0 : tensor<f32> 13 // CHECK-SAME: %[[ARG:.*]]: memref<f32>) -> memref<f32> { 14 // CHECK: %[[T1:.*]] = tensor_load %[[ARG]] : memref<f32> 15 // CHECK: %[[M1:.*]] = tensor_to_memref %[[T1]] : memref<f32> 16 // CHECK: br ^bb1(%[[M1]] : memref<f32>) [all …]
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/compile_mlir_util/ |
D | add.mlir | 7 func @main(%arg0: tensor<f32>, %arg1: tensor<f32>) -> tensor<f32> { 8 %0 = "tf.AddV2"(%arg0, %arg1) : (tensor<f32>, tensor<f32>) -> tensor<f32> 9 return %0 : tensor<f32> 14 // CHECK: ENTRY %main.{{[0-9]+}} ([[ARG0:.*]]: f32[], [[ARG1:.*]]: f32[]) -> (f32[]) { 15 // CHECK-NEXT: %[[ARG0]] = f32[] parameter(0) 16 // CHECK-NEXT: %[[ARG1]] = f32[] parameter(1) 17 // CHECK-NEXT: [[ADD:%.*]] = f32[] add(f32[] %[[ARG0]], f32[] %[[ARG1]]) 18 // CHECK-NEXT: ROOT %tuple.{{[0-9]+}} = (f32[]) tuple(f32[] [[ADD]]) 22 // CHECK-NEXT: // XlaInputShape f32[] 23 // CHECK-NEXT: // XlaInputShape f32[] [all …]
|
/external/llvm-project/mlir/test/Dialect/Linalg/ |
D | forward-vector-transfers.mlir | 10 func @testAllocRead(%in: memref<? x f32>) -> vector<32 x f32> { 12 %f0 = constant 0.0: f32 13 %alloc = alloc() : memref<32 x f32> 14 %subview = subview %alloc[0][16][1] : memref<32 x f32> to memref<16 x f32> 15 linalg.copy(%in, %subview): memref<? x f32>, memref<16 x f32> 16 %0 = vector.transfer_read %alloc[%c0], %f0 {masked = [false]} : memref<32 x f32>, vector<32 x f32> 17 dealloc %alloc : memref<32 x f32> 18 return %0: vector<32 x f32> 28 func @testAllocFillRead(%in: memref<? x f32>) -> vector<32 x f32> { 30 %f0 = constant 0.0: f32 [all …]
|
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/tests/ |
D | tf_abi_knowledge.mlir | 28 … // ABI-SAME: %[[ARG0:.*]]: !llvm.ptr<f32>, %[[ARG1:.*]]: !llvm.ptr<f32> {llvm.align = 16 : index}, 29 …[ARG3:.*]]: i64, %[[ARG4:.*]]: i64, %[[ARG5:.*]]: !llvm.ptr<f32>, %[[ARG6:.*]]: !llvm.ptr<f32> {ll… 31 …<f32>, %[[ARG1:.*]]: !llvm.ptr<f32>, %[[ARG2:.*]]: i64, %[[ARG3:.*]]: i64, %[[ARG4:.*]]: i64, %[[A… 32 …rnel(%arg0: !llvm.ptr<f32>, %arg1: !llvm.ptr<f32>, %arg2: i64, %arg3: i64, %arg4: i64, %arg5: !llv… 36 %0 = llvm.mlir.undef : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<1 x i64>, array<1 x i64>)> 39 …%1 = llvm.insertvalue %arg0, %0[0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<1 x i64>, array<… 41 …%2 = llvm.insertvalue %arg1, %1[1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<1 x i64>, array<… 44 …%3 = llvm.insertvalue %arg2, %2[2] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<1 x i64>, array<… 46 …%4 = llvm.insertvalue %arg3, %3[3, 0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<1 x i64>, arr… 49 …%5 = llvm.insertvalue %arg4, %4[4, 0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<1 x i64>, arr… [all …]
|