| /external/llvm-project/mlir/test/Dialect/Standard/ |
| D | func-bufferize.mlir | 5 // CHECK: %[[TENSOR:.*]] = tensor_load %[[ARG]] : memref<f32>
6 // CHECK: %[[MEMREF:.*]] = tensor_to_memref %[[TENSOR]] : memref<f32>
38 // CHECK: %[[TENSOR:.*]] = tensor_load %[[ARG]] : memref<f32>
39 // CHECK: %[[MEMREF:.*]] = tensor_to_memref %[[TENSOR]] : memref<f32>
49 // CHECK: %[[TENSOR:.*]] = "test.source"() : () -> tensor<f32>
50 // CHECK: %[[MEMREF:.*]] = tensor_to_memref %[[TENSOR]] : memref<f32>
|
| D | bufferize.mlir | 4 // CHECK-SAME: %[[TENSOR:.*]]: tensor<f32>,
6 // CHECK: %[[MEMREF:.*]] = tensor_to_memref %[[TENSOR]] : memref<f32>
65 // CHECK-SAME: %[[TENSOR:.*]]: tensor<?xf32>,
67 // CHECK: %[[MEMREF:.*]] = tensor_to_memref %[[TENSOR]] : memref<?xf32>
91 // CHECK-SAME: %[[TENSOR:.*]]: tensor<?xindex>) -> tensor<2xindex> {
92 // CHECK: %[[MEMREF:.*]] = tensor_to_memref %[[TENSOR]]
102 // CHECK-SAME: %[[TENSOR:.*]]: tensor<*xf32>) -> tensor<2xf32> {
103 // CHECK: %[[MEMREF:.*]] = tensor_to_memref %[[TENSOR]] : memref<*xf32>
113 // CHECK-SAME: %[[TENSOR:.*]]: tensor<2xf32>) -> tensor<*xf32> {
114 // CHECK: %[[MEMREF:.*]] = tensor_to_memref %[[TENSOR]] : memref<2xf32>
|
| D | canonicalize.mlir | 5 // CHECK-SAME: %[[TENSOR:.*]]: tensor<?xf32>) -> tensor<?x…
6 // CHECK: return %[[TENSOR]]
26 // CHECK: %[[TENSOR:.*]] = tensor_load %[[MEMREF_ADDRSPACE2]] : memref<?xf32, 2>
27 // CHECK: %[[MEMREF_ADDRSPACE7:.*]] = tensor_to_memref %[[TENSOR]] : memref<?xf32, 7>
52 // CHECK-SAME: %[[TENSOR:[0-9a-z]+]]: tensor<?x?xf32>
53 // CHECK: %[[RES:.*]] = extract_element %[[TENSOR]][%[[IDX0]], %[[IDX1]]]
|
| D | tensor-constant-bufferize.mlir | 10 // CHECK: %[[TENSOR:.*]] = tensor_load %[[MEMREF]]
12 // CHECK: return %[[TENSOR]]
|
| /external/llvm-project/mlir/test/Dialect/SCF/ |
| D | bufferize.mlir | 27 // CHECK-SAME: %[[TENSOR:.*]]: tensor<f32>,
30 // CHECK: %[[MEMREF:.*]] = tensor_to_memref %[[TENSOR]] : memref<f32>
63 // CHECK-SAME: %[[TENSOR:.*]]: tensor<f32>,
65 // CHECK: %[[MEMREF:.*]] = tensor_to_memref %[[TENSOR]] : memref<f32>
72 // CHECK: %[[TENSOR:.*]] = tensor_load %[[RESULT:.*]] : memref<f32>
73 // CHECK: return %[[TENSOR]] : tensor<f32>
|
| /external/libaom/libaom/av1/encoder/ |
| D | cnn.c | 46 } TENSOR; typedef
48 static void init_tensor(TENSOR *tensor) { memset(tensor, 0, sizeof(*tensor)); } in init_tensor()
50 static void free_tensor(TENSOR *tensor) { in free_tensor()
58 static void realloc_tensor(TENSOR *tensor, int channels, int width, in realloc_tensor()
75 static void copy_tensor(const TENSOR *src, int copy_channels, int dst_offset, in copy_tensor()
76 TENSOR *dst) { in copy_tensor()
96 static void assign_tensor(TENSOR *tensor, float *buf[CNN_MAX_CHANNELS], in assign_tensor()
110 static void swap_tensor(TENSOR *t1, TENSOR *t2) { in swap_tensor()
111 TENSOR t = *t1; in swap_tensor()
118 static void concat_tensor(const TENSOR *src, TENSOR *dst) { in concat_tensor()
[all …]
|
| /external/llvm-project/mlir/test/Dialect/Shape/ |
| D | bufferize.mlir | 12 // CHECK: %[[TENSOR:.*]] = tensor_load %[[MEMREF:.*]] : memref<2xf16>
13 // CHECK: "test.sink"(%[[TENSOR]]) : (tensor<2xf16>) -> ()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/python/ |
| D | tfr_gen.py | 62 TENSOR = 1 variable in TFRTypes
120 return TFRTypes.TENSOR
411 qual_names.QN(n.value): {TFRTypes.TENSOR}
426 TFRTypes.TENSOR_LIST, TFRTypes.TENSOR, TFRTypes.ATTR
510 elif value in (TFRTypes.TENSOR_LIST, TFRTypes.TENSOR):
512 return {TFRTypes.TENSOR}
531 if TFRTypes.TENSOR in elt_types:
549 if TFRTypes.TENSOR in all_elt_types:
662 return casted, TFRTypes.TENSOR
687 return cst_tensor, TFRTypes.TENSOR
[all …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | op_def_util.h | 40 TENSOR, // "tensor" (tf.TensorProto) enumerator
|
| D | op_def_util.cc | 57 {"tensor", AttributeType::TENSOR}, in AttributeTypeNameMap()
228 case AttributeType::TENSOR: in ConvertAttrOrNull()
|
| /external/llvm-project/mlir/test/Dialect/Linalg/ |
| D | bufferize.mlir | 13 // CHECK-SAME: %[[TENSOR:.*]]: tensor<4xf32>) -> tensor<4xf32> {
14 // CHECK: %[[MEMREF:.*]] = tensor_to_memref %[[TENSOR]] : memref<4xf32>
|
| /external/llvm-project/mlir/test/Transforms/ |
| D | canonicalize.mlir | 1056 // CHECK-SAME: %[[IDX:.*]]: index, %[[TENSOR:.*]]: tensor<*xf32>
1059 // CHECK-NEXT: %[[RES:.*]] = dim %[[TENSOR]], %[[IDX]]
1073 // CHECK-SAME: %[[IDX0:.*]]: index, %[[IDX1:.*]]: index, %[[TENSOR:.*]]: tensor<*xf32>
1076 // CHECK-NEXT: %[[DIM0:.*]] = dim %[[TENSOR]], %[[IDX0]]
1077 // CHECK-NEXT: %[[DIM1:.*]] = dim %[[TENSOR]], %[[IDX1]]
1209 // CHECK-SAME: %[[TENSOR:.*]]: tensor<*xf32>
1215 // CHECK-NEXT: extract_element %[[TENSOR]][%[[C0]]]
|
| /external/llvm-project/mlir/test/IR/ |
| D | core-ops.mlir | 886 // CHECK: %[[TENSOR:.*]] = tensor_load %[[MEMREF:.*]] : memref<4x4xi32>
888 // CHECK: tensor_store %[[TENSOR]], %[[MEMREF]] : memref<4x4xi32>
895 // CHECK: %[[TENSOR:.*]] = tensor_load %[[MEMREF:.*]] : memref<*xi32>
897 // CHECK: tensor_store %[[TENSOR]], %[[MEMREF]] : memref<*xi32>
|
| /external/llvm-project/mlir/test/Conversion/ShapeToStandard/ |
| D | shape-to-standard.mlir | 110 // CHECK: %[[TENSOR:.*]] = tensor_from_elements : tensor<0xindex>
111 // CHECK: %[[RESULT:.*]] = tensor_cast %[[TENSOR]] : tensor<0xindex> to tensor<?xindex>
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | decompose_resource_ops.mlir | 483 …// CHECK: [[TENSOR:%.+]] = "tf.TensorScatterUpdate"([[READ]], [[INDEX]], [[UPDATE]]) : (tensor<*xi…
484 // CHECK: "tf.AssignVariableOp"([[VAR]], [[TENSOR]])
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 1926 // CHECK: [[TENSOR:%.+]] = shape.to_extent_tensor [[SHAPE]]
1927 // CHECK: [[CAST:%.+]] = index_cast [[TENSOR]]
1937 // CHECK: [[TENSOR:%.+]] = shape.to_extent_tensor [[SHAPE]]
1938 // CHECK: [[CAST:%.+]] = index_cast [[TENSOR]]
1948 // CHECK: [[TENSOR:%.+]] = shape.to_extent_tensor [[SHAPE]]
1949 // CHECK: [[CAST:%.+]] = index_cast [[TENSOR]]
|