Home
last modified time | relevance | path

Searched refs:memref (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlhlo_ops.mlir3 func @enforce_same_shape(%arg0: memref<1xf32>, %arg1: memref<2xf32>) -> () {
5 "xla_lhlo.tanh"(%arg0, %arg1) : (memref<1xf32>, memref<2xf32>) -> ()
12 func @add_memrefs(%arg0: memref<1xi32>, %arg1: memref<1xi32>, %arg_out: memref<1xi32>) -> () {
13 "xla_lhlo.add"(%arg0, %arg1, %arg_out) : (memref<1xi32>, memref<1xi32>, memref<1xi32>) -> ()
20 func @abs_memref(%in: memref<10xf32>, %out: memref<10xf32>) -> () {
21 "xla_lhlo.abs"(%in, %out) : (memref<10xf32>, memref<10xf32>) -> ()
28 func @convert_memref(%in: memref<10xf32>, %out: memref<10xf32>) -> () {
29 "xla_lhlo.convert"(%in, %out) : (memref<10xf32>, memref<10xf32>) -> ()
36 func @exp_memref(%in: memref<10xf32>, %out: memref<10xf32>) -> () {
37 "xla_lhlo.exp"(%in, %out) : (memref<10xf32>, memref<10xf32>) -> ()
[all …]
Dlhlo-legalize-to-affine.mlir5 func @min_op(%lhs: memref<4x3x2x1xf32>, %rhs: memref<4x3x2x1xf32>,
6 %result: memref<4x3x2x1xf32>) -> () {
11 // CHECK-NEXT: %[[LHS:.*]] = load %{{.*}}[%[[I]], %[[J]], %[[K]], %[[L]]] : memref<4x3x2x1xf32>
12 // CHECK-NEXT: %[[RHS:.*]] = load %{{.*}}[%[[I]], %[[J]], %[[K]], %[[L]]] : memref<4x3x2x1xf32>
15 // CHECK-NEXT: store %[[MIN]], %{{.*}}[%[[I]], %[[J]], %[[K]], %[[L]]] : memref<4x3x2x1xf32>
18 (memref<4x3x2x1xf32>, memref<4x3x2x1xf32>, memref<4x3x2x1xf32>) -> ()
24 func @float_add_op(%lhs: memref<7xf32>, %rhs: memref<7xf32>,
25 %result: memref<7xf32>) -> () {
28 : (memref<7xf32>, memref<7xf32>, memref<7xf32>) -> ()
32 func @int_add_op(%lhs: memref<7xi32>, %rhs: memref<7xi32>,
[all …]
Dlhlo-legalize-to-linalg.mlir5 func @element_wise(%lhs: memref<2x2xf32>, %rhs: memref<2x2xf32>,
6 %result: memref<2x2xf32>) {
8 : (memref<2x2xf32>, memref<2x2xf32>, memref<2x2xf32>) -> ()
19 func @element_wise_with_dynamic_shape(%lhs: memref<?x?xf32>, %rhs: memref<?x?xf32>,
20 %result: memref<?x?xf32>) {
22 : (memref<?x?xf32>, memref<?x?xf32>, memref<?x?xf32>) -> ()
33 func @element_wise_scalar(%lhs: memref<f32>, %rhs: memref<f32>,
34 %result: memref<f32>) {
41 : (memref<f32>, memref<f32>, memref<f32>) -> ()
48 func @minf(%lhs: memref<2x2xf32>, %rhs: memref<2x2xf32>,
[all …]
Dhlo-legalize-to-lhlo.mlir4 func @attrs_copy(%operand: memref<2x2xf32>, %result: memref<2x2xf32>) {
5 %tensor_operand = tensor_load %operand : memref<2x2xf32>
10 tensor_store %tensor_result, %result : memref<2x2xf32>
16 // CHECK-NEXT: %[[MAX_RESULT:.*]] = alloc() {temp = true} : memref<4xf32>
20 // CHECK-NEXT: dealloc %[[MAX_RESULT]] : memref<4xf32>
27 // CHECK-NEXT: %[[MUL_RESULT:.*]] = alloc() {temp = true} : memref<4xf32>
28 // CHECK-NEXT: %[[SUB_RESULT:.*]] = alloc() {temp = true} : memref<4xf32>
29 // CHECK-NEXT: %[[MIN_RESULT:.*]] = alloc() {temp = true} : memref<4xf32>
30 // CHECK-NEXT: %[[ADD_RESULT:.*]] = alloc() {temp = true} : memref<4xf32>
31 // CHECK-NEXT: %[[MAX_RESULT:.*]] = alloc() {temp = true} : memref<4xf32>
[all …]
Dlhlo-legalize-to-gpu.mlir3 func @reduce(%arg: memref<100x10xf32>,
4 %init: memref<f32>,
5 %result: memref<100xf32>) {
7 ^bb0(%lhs: memref<f32>, %rhs: memref<f32>, %res: memref<f32>):
9 : (memref<f32>, memref<f32>, memref<f32>) -> ()
12 : (memref<100x10xf32>, memref<f32>, memref<100xf32>) -> ()
17 // CHECK: func @reduce(%[[ARG0:.*]]: memref<100x10xf32>, %[[ARG1:.*]]: memref<f32>, %[[ARG2:.*]]: m…
21 // CHECK: %[[ACC:.*]] = load %[[ARG1]][] : memref<f32>
22 // CHECK: store %[[ACC]], %[[ARG2]][%[[IDX:.*]]] : memref<100xf32>
27 // CHECK: %[[LHS:.*]] = linalg.slice %[[ARG2]][%[[IDX]]] : memref<100xf32>, index, memref<f32, #map…
[all …]
Dlhlo-fuse-linalg.mlir5 func @fusion(%multiplier: memref<2x2xf32>, %summand_1: memref<2x2xf32>,
6 %summand_2: memref<2x2xf32>, %result: memref<2x2xf32>) {
7 %temp_result = alloc() {temp = true} : memref<2x2xf32>
12 } : memref<2x2xf32>, memref<2x2xf32>, memref<2x2xf32>
17 } : memref<2x2xf32>, memref<2x2xf32>, memref<2x2xf32>
18 dealloc %temp_result : memref<2x2xf32>
31 func @fusion_of_three(%arg0: memref<100x10xf32>,
32 %arg1: memref<100xf32>,
33 %arg2: memref<100x10xf32>) {
34 %0 = alloc() {temp = true} : memref<100x10xf32>
[all …]
/external/autotest/client/site_tests/platform_MemCheck/
Dplatform_MemCheck.py31 memref = 986392
34 memref = 700000
59 memref = max(memref, phy_size - os_reserve)
61 ref = {'MemTotal': memref,
68 board, phy_size, memref)
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/
Dkernel_lowering.cc161 GetAllocOp(storeOp.memref(), &memrefToAllocOp); in runOnFunction()
163 GetAllocOp(loadOp.memref(), &memrefToAllocOp); in runOnFunction()
182 mlir::Operation* SearchAllocOp(mlir::Value memref) { in SearchAllocOp()
183 mlir::Operation* defOp = memref.getDefiningOp(); in SearchAllocOp()
195 mlir::Value memref, in GetAllocOp()
197 auto allocOpIt = memrefToAllocOp->find(memref); in GetAllocOp()
201 auto allocOp = SearchAllocOp(memref); in GetAllocOp()
202 memrefToAllocOp->insert({memref, allocOp}); in GetAllocOp()
/external/u-boot/drivers/tee/
Dsandbox.c170 name = params[0].u.memref.shm->addr; in ta_avb_invoke_func()
172 value = params[1].u.memref.shm->addr; in ta_avb_invoke_func()
173 value_sz = params[1].u.memref.size; in ta_avb_invoke_func()
194 name = params[0].u.memref.shm->addr; in ta_avb_invoke_func()
196 value = params[1].u.memref.shm->addr; in ta_avb_invoke_func()
197 value_sz = params[1].u.memref.size; in ta_avb_invoke_func()
/external/u-boot/common/
Davb_verify.c890 param[0].u.memref.shm = shm_name; in read_persistent_value()
891 param[0].u.memref.size = name_size; in read_persistent_value()
893 param[1].u.memref.shm = shm_buf; in read_persistent_value()
894 param[1].u.memref.size = buffer_size; in read_persistent_value()
901 if (param[1].u.memref.size > buffer_size) { in read_persistent_value()
906 *out_num_bytes_read = param[1].u.memref.size; in read_persistent_value()
955 param[0].u.memref.shm = shm_name; in write_persistent_value()
956 param[0].u.memref.size = name_size; in write_persistent_value()
958 param[1].u.memref.shm = shm_buf; in write_persistent_value()
959 param[1].u.memref.size = value_size; in write_persistent_value()
/external/u-boot/drivers/tee/optee/
Dcore.c188 mp->u.rmem.shm_ref = (ulong)p->u.memref.shm; in to_msg_param()
189 mp->u.rmem.size = p->u.memref.size; in to_msg_param()
190 mp->u.rmem.offs = p->u.memref.shm_offs; in to_msg_param()
229 p->u.memref.size = mp->u.rmem.size; in from_msg_param()
233 p->u.memref.shm_offs = 0; in from_msg_param()
234 p->u.memref.shm = NULL; in from_msg_param()
237 p->u.memref.shm_offs = mp->u.rmem.offs; in from_msg_param()
238 p->u.memref.shm = shm; in from_msg_param()
/external/u-boot/include/
Dtee.h127 struct tee_param_memref memref; member
/external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/experimental/conv_emitter/
Dconv_emitter.cc154 void SetMemRef(mlir::Operation* op, mlir::Value memref) { in SetMemRef() argument
156 load.setMemRef(memref); in SetMemRef()
158 store.setMemRef(memref); in SetMemRef()