Home
last modified time | relevance | path

Searched refs:SHAPE (Results 1 – 25 of 53) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dreplicate_invariant_op_hoisting.mlir14 // CHECK: %[[SHAPE:[0-9]*]] = "tf.Shape"(%[[ARG_0]])
16 // CHECK: %[[OP_A:[0-9]*]] = "tf.opA"(%[[SHAPE]], %[[RI]])
17 // CHECK: tf_device.return %[[SHAPE]], %[[OP_A]]
30 // CHECK: %[[SHAPE:[0-9]*]] = "tf.Shape"(%[[ARG_1]])
32 // CHECK: tf_device.return %[[SHAPE]]
64 // CHECK: %[[SHAPE:[0-9]*]] = "tf.Shape"(%[[ARG_2]])
66 // CHECK: %[[OP_A:[0-9]*]] = "tf.opA"(%[[SHAPE]], %[[RJ]])
67 // CHECK: tf_device.return %[[SHAPE]], %[[OP_A]]
99 // CHECK: %[[SHAPE:[0-9]*]] = "tf.Shape"(%[[READ_VAR]])
101 // CHECK: tf_device.return %[[SHAPE]]
[all …]
Dlower_tf.mlir6 …// CHECK-NEXT: %[[SHAPE:.*]] = "tf.Const"() {value = dense<[5, 1]> : tensor<2xi32>} : () -> tensor…
7 …// CHECK-NEXT: %[[INDICES:.*]] = "tf.Reshape"(%arg0, %[[SHAPE]]) : (tensor<5xi32>, tensor<2xi32>) …
532 // CHECK: %[[SHAPE:.*]] = "tf.Shape"(%arg0) : (tensor<*xi32>) -> tensor<?xi64>
533 // CHECK: "tf.BroadcastTo"(%[[ZERO]], %[[SHAPE]]) : (tensor<i32>, tensor<?xi64>) -> tensor<*xi32>
549 // CHECK: %[[SHAPE:.*]] = "tf.Shape"(%arg0) : (tensor<*xi32>) -> tensor<?xi64>
550 // CHECK: "tf.BroadcastTo"(%[[ONE]], %[[SHAPE]]) : (tensor<i32>, tensor<?xi64>) -> tensor<*xi32>
617 …// CHECK-DAG: %[[SHAPE:.*]] = "tf.Const"() {value = dense<[-1, 2]> : tensor<2xi64>} : () -> tensor…
618 …// CHECK-DAG: %[[INP0:.*]] = "tf.Reshape"(%arg0, %[[SHAPE]]) : (tensor<2xf32>, tensor<2xi64>) -> t…
620 …// CHECK-DAG: %[[INP1:.*]] = "tf.Reshape"(%arg1, %[[SHAPE]]) : (tensor<2x2x2xf32>, tensor<2xi64>) …
985 // CHECK: %[[SHAPE:.*]] = "tf.Shape"(%arg0) : (tensor<1x?x2x3xf32>) -> tensor<4xi32>
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/tests/
Dtf_abi_knowledge.mlir2 …-propagate-shape-knowledge-to-kernels -split-input-file | FileCheck %s --check-prefixes=CHECK,SHAPE
31 …// SHAPE-SAME: %[[ARG0:.*]]: !llvm.ptr<f32>, %[[ARG1:.*]]: !llvm.ptr<f32>, %[[ARG2:.*]]: i64, %[[A…
38 // SHAPE-NEXT: llvm.insertvalue %[[ARG0]]
43 // SHAPE-NEXT: llvm.insertvalue %[[ARG2]]
48 // SHAPE-NEXT: llvm.insertvalue %[[ARG4]]
53 // SHAPE-NEXT: llvm.insertvalue %[[ARG5]]
58 // SHAPE-NEXT: llvm.insertvalue %[[ARG7]]
61 // SHAPE-NEXT: llvm.insertvalue %[[ARG3]]
64 // SHAPE-NEXT: llvm.insertvalue %[[ARG4]]
111 …// SHAPE: llvm.mlir.undef : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<1 x i64>, array<1 x i64>)>
[all …]
Dprint_memrefs.mlir31 // CHECK: [[SHAPE:%.*]] = alloca({{%.*}}) : memref<?xindex>
44 // CHECK: [[SHAPE_I64:%.*]] = index_cast [[SHAPE]]
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dhlo-legalize-to-lhlo-unranked.mlir19 // CHECK-SAME: ([[ARG:%.*]]: memref<*xf32>, [[SHAPE:%.*]]: memref<1xi32>)
20 // CHECK-NEXT: memref_reshape [[ARG]]([[SHAPE]])
32 // CHECK-SAME: ([[ARG:%.*]]: memref<?xf32>, [[SHAPE:%.*]]: memref<?xi32>)
33 // CHECK-NEXT: memref_reshape [[ARG]]([[SHAPE]])
Dmhlo_infer_shape_type_methods.mlir12 // CHECK: %[[SHAPE:.*]] = tensor.from_elements %[[C2]], %[[DIM]] : tensor<2xi64>
13 // CHECK: return %[[SHAPE]] : tensor<2xi64>
29 // CHECK: %[[SHAPE:.*]] = tensor.from_elements %[[C2]], %[[DIM]] : tensor<2xi64>
30 // CHECK: return %[[SHAPE]] : tensor<2xi64>
Dhlo-transform-unranked.mlir30 // CHECK-NEXT: %[[SHAPE:.*]] = shape.shape_of %[[A]] : tensor<*xf32> -> tensor<?xindex>
31 // CHECK-NEXT: %[[NUM_ELEMENTS:.*]] = shape.num_elements %[[SHAPE]]
35 …// CHECK-NEXT: %[[B:.*]] = "mhlo.dynamic_reshape"(%[[FLAT_B]], %[[SHAPE]]) : (tensor<?xf32>, tenso…
72 // CHECK: %[[SHAPE:.*]] = shape.any %[[SHAPE_A]], %[[SHAPE_B]]
73 // CHECK: %[[NUM_ELEMENTS:.*]] = shape.num_elements %[[SHAPE]]
78 …// CHECK: %[[RESULT:.*]] = "mhlo.dynamic_reshape"(%[[FLAT_RESULT]], %[[SHAPE]]) : (tensor<?xf32>, …
89 // CHECK: %[[SHAPE:.*]] = shape.shape_of %[[A]] : tensor<*xf32> -> tensor<?xindex>
90 // CHECK: %[[NUM_ELEMENTS:.*]] = shape.num_elements %[[SHAPE]]
94 …// CHECK: %[[B:.*]] = "mhlo.dynamic_reshape"(%[[FLAT_B]], %[[SHAPE]]) : (tensor<?xf32>, tensor<?xi…
Dhlo-legalize-to-lhlo.mlir153 // CHECK: %[[SHAPE:.*]] = tensor.from_elements
161 // CHECK: %[[EL0:.*]] = tensor.extract %[[SHAPE]]{{\[}}%[[C0]]] : tensor<3xi64>
163 // CHECK: %[[EL1:.*]] = tensor.extract %[[SHAPE]]{{\[}}%[[C1]]] : tensor<3xi64>
170 // CHECK: %[[EL2:.*]] = tensor.extract %[[SHAPE]]{{\[}}%[[C2]]] : tensor<3xi64>
475 // CHECK: %[[SHAPE:.*]] = tensor.from_elements %[[IC0]], %[[IC1]] : tensor<2xi64>
476 // CHECK: %[[EE0:.*]] = tensor.extract %[[SHAPE]][%[[C0]]] : tensor<2xi64>
478 // CHECK: %[[EE1:.*]] = tensor.extract %[[SHAPE]][%[[C1]]] : tensor<2xi64>
499 // CHECK: %[[SHAPE:.*]] = tensor.from_elements %[[IC0]], %[[IC1]] : tensor<2xi64>
500 // CHECK: %[[EE0:.*]] = tensor.extract %[[SHAPE]][%[[C0]]] : tensor<2xi64>
502 // CHECK: %[[EE1:.*]] = tensor.extract %[[SHAPE]][%[[C1]]] : tensor<2xi64>
/external/llvm-project/mlir/test/Dialect/Standard/
Dexpand-ops.mlir93 // CHECK-SAME: [[SHAPE:%.*]]: memref<3xi32>) -> memref<?x?x?xf32> {
97 // CHECK: [[DIM_2:%.*]] = load [[SHAPE]]{{\[}}[[C2]]] : memref<3xi32>
102 // CHECK: [[DIM_1:%.*]] = load [[SHAPE]]{{\[}}[[C1_]]] : memref<3xi32>
107 // CHECK: [[DIM_0:%.*]] = load [[SHAPE]]{{\[}}[[C0]]] : memref<3xi32>
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/
Dlower-static-tensor-list.mlir56 // CHECK-NEXT: [[SHAPE:%.*]] = "tf.Shape"(%arg0) : (tensor<?x3xf32>) -> tensor<?xi32>
57 // CHECK-NEXT: [[RESHAPE:%.*]] = "tf.Reshape"(%arg0, [[SHAPE]]) : (tensor<?x3xf32>, tensor<?xi32>)…
160 // CHECK-DAG: [[SHAPE:%.*]] = "tf.Concat"([[ZERO2]], [[DIM0]], %arg0) : (tensor<i32>, tensor<1xi32…
162 // CHECK: [[LIST:%.*]] = "tf.Fill"([[SHAPE]], [[VALUES]]) : (tensor<4xi32>, tensor<f32>) -> te…
189 // CHECK-DAG: [[SHAPE:%.*]] = "tf.Concat"([[ZERO2]], [[DIM0]], [[ELEMENT_SHAPE]]) : (tensor<i32>, …
191 // CHECK: [[LIST:%.*]] = "tf.Fill"([[SHAPE]], [[VALUES]]) : (tensor<3xi32>, tensor<f32>) -> te…
204 // CHECK-DAG: [[SHAPE:%[0-9]+]] = "tf.Shape"(%arg3) : (tensor<2xf32>) -> tensor<?xi32>
208 // CHECK-DAG: [[FINAL_SHAPE:%[0-9]+]] = "tf.Concat"([[CST0]], [[EXPAND_DIM]], [[SHAPE]]) : (tensor…
222 // CHECK-DAG: [[SHAPE:%.*]] = "tf.Concat"([[ZERO]], [[DIM0]], [[ELEM_SHAPE]]) : (tensor<i32>, tens…
224 // CHECK: [[LIST:%.*]] = "tf.Fill"([[SHAPE]], [[VALUES]]) : (tensor<4xi32>, tensor<f32>) -> te…
[all …]
/external/tensorflow/tensorflow/python/debug/lib/
Ddumping_callback.py411 debug_event_pb2.TensorDebugMode.SHAPE):
419 tensor_debug_mode == debug_event_pb2.TensorDebugMode.SHAPE and
499 debug_event_pb2.TensorDebugMode.SHAPE,
526 elif tensor_debug_mode == debug_event_pb2.TensorDebugMode.SHAPE:
824 debug_event_pb2.TensorDebugMode.SHAPE,
Ddebug_v2_ops_test.py339 debug_event_pb2.TensorDebugMode.SHAPE,
556 tensor_debug_mode=(debug_event_pb2.TensorDebugMode.SHAPE),
571 tensor_debug_mode=(debug_event_pb2.TensorDebugMode.SHAPE),
603 tensor_debug_mode=(debug_event_pb2.TensorDebugMode.SHAPE),
/external/tensorflow/tensorflow/python/framework/
Dop_def_util.h39 SHAPE, // "shape" (tf.TensorShape) enumerator
Dop_def_util.cc55 {"shape", AttributeType::SHAPE}, in AttributeTypeNameMap()
226 case AttributeType::SHAPE: in ConvertAttrOrNull()
/external/libxkbcommon/src/xkbcomp/
Dparser.y116 SHAPE 29
501 ShapeDecl : SHAPE String OBRACE OutlineList CBRACE SEMI
503 | SHAPE String OBRACE CoordList CBRACE SEMI
609 | SHAPE
Dkeywords.gperf51 shape, SHAPE
Dkeywords.c306 {(int)(size_t)&((struct stringpool_t *)0)->stringpool_str30, SHAPE}, in keyword_gperf_lookup()
/external/llvm-project/mlir/test/Conversion/ShapeToStandard/
Dshape-to-standard.mlir31 // CHECK-SAME: (%[[SHAPE:.*]]: tensor<?xindex>) -> index
34 // CHECK: %[[RESULT:.*]] = dim %[[SHAPE]], %[[C0]]
168 // CHECK-SAME: (%[[SHAPE:.*]]: tensor<?xindex>) -> index
181 // CHECK-NEXT: %[[RANK:.*]] = dim %[[SHAPE]], %[[C0]] : tensor<?xindex>
183 // CHECK-NEXT: %[[EXTENT:.*]] = extract_element %[[SHAPE]][%[[I]]]
207 // CHECK: %[[SHAPE:.*]] = dynamic_tensor_from_elements %[[RANK]] {
/external/libxkbcommon/config/libxkbcommon.so.0.0.0.p/
Dparser.h80 SHAPE = 29, /* SHAPE */ enumerator
/external/tensorflow/tensorflow/compiler/mlir/tfr/tests/
Dend2end.mlir30 // CHECK-NEXT: %[[SHAPE:.*]] = "tf.RiscShape"(%arg0) {T = i32} : (tensor<2x3xf32>) -> tensor<*xi32>
31 // CHECK-NEXT: %[[ALPHA1:.*]] = "tf.RiscBroadcast"(%[[ALPHA]], %[[SHAPE]]) : (tensor<f32>, tensor<*…
43 // CHECK-NEXT: %[[SHAPE:.*]] = "tf.RiscShape"(%arg0) {T = i32} : (tensor<2x3xf32>) -> tensor<*xi32>
44 // CHECK-NEXT: %[[ALPHA1:.*]] = "tf.RiscBroadcast"(%[[ALPHA]], %[[SHAPE]]) : (tensor<f32>, tensor<*…
/external/llvm-project/mlir/test/Dialect/Shape/
Dcanonicalize.mlir608 // CHECK-SAME: (%[[SHAPE:.*]]: !shape.shape) -> !shape.size
610 // CHECK: %[[RESULT:.*]] = shape.rank %[[SHAPE]]
632 // CHECK-SAME: (%[[SHAPE:.*]]: tensor<?xindex>) -> index
634 // CHECK: %[[RESULT:.*]] = shape.rank %[[SHAPE]] : tensor<?xindex> -> index
670 // CHECK: %[[SHAPE:.*]] = shape.shape_of %[[ARG]] : tensor<*xf32> -> tensor<?xindex>
671 // CHECK: %[[SIZE:.*]] = shape.rank %[[SHAPE]]
/external/lottie/lottie/src/main/java/com/airbnb/lottie/model/layer/
DLayer.java24 SHAPE, enumConstant
/external/tensorflow/tensorflow/compiler/mlir/tfr/python/
Dtfr_gen.py66 SHAPE = 5 # shape -> !shape.shape variable in TFRTypes
84 return TFRTypes.SHAPE
382 if value in (TFRTypes.SHAPE, TFRTypes.TF_TENSOR_SHAPE_FUNC):
729 if node_type == TFRTypes.SHAPE:
736 return (ssa_value, TFRTypes.SHAPE)
747 if ty == TFRTypes.SHAPE and node.attr == 'as_list':
/external/llvm-project/flang/include/flang/Evaluate/
Dconstant.h69 Constant<SubscriptInteger> SHAPE() const;
Dshape.h104 return AsShape(c.SHAPE()); in operator()

123