Home
last modified time | relevance | path

Searched refs:comparison_direction (Results 1 – 25 of 74) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/tests/
Dverify_select_and_scatter.mlir13 comparison_direction = #stablehlo<comparison_direction GE>
41 comparison_direction = #stablehlo<comparison_direction GE>}
71 comparison_direction = #stablehlo<comparison_direction GE>
98 comparison_direction = #stablehlo<comparison_direction GE>
125 comparison_direction = #stablehlo<comparison_direction GE>
153 comparison_direction = #stablehlo<comparison_direction GE>
180 comparison_direction = #stablehlo<comparison_direction GE>
208 comparison_direction = #stablehlo<comparison_direction GE>
236 comparison_direction = #stablehlo<comparison_direction GE>
262 comparison_direction = #stablehlo<comparison_direction GE>
[all …]
Dprint_while_invalid.mlir14 …%4 = "stablehlo.compare"(%arg1, %3) {comparison_direction = "LT"} : (tensor<1xi32>, tensor<1xi32>)…
37 …%4 = "stablehlo.compare"(%arg1, %3) {comparison_direction = "LT"} : (tensor<1xi32>, tensor<1xi32>)…
60 …%4 = "stablehlo.compare"(%arg1, %3) {comparison_direction = "LT"} : (tensor<1xi32>, tensor<1xi32>)…
83 …%4 = "stablehlo.compare"(%arg1, %3) {comparison_direction = "LT"} : (tensor<1xi32>, tensor<1xi32>)…
123 …%4 = "stablehlo.compare"(%arg1, %3) {comparison_direction = "LT"} : (tensor<1xi32>, tensor<1xi32>)…
165 …%4 = "stablehlo.compare"(%arg1, %3) {comparison_direction = "LT"} : (tensor<1xi32>, tensor<1xi32>)…
186 …%4 = "stablehlo.compare"(%arg1, %3) {comparison_direction = "LT"} : (tensor<1xi32>, tensor<1xi32>)…
209 …%4 = "stablehlo.compare"(%arg1, %3) {comparison_direction = "LT"} : (tensor<1xi32>, tensor<1xi32>)…
Dprint_while.mlir25 …%1 = "stablehlo.compare"(%iterArg, %iterArg_4) {comparison_direction = #stablehlo<comparison_direc…
Dinfer_chlo.mlir30 …%0 = chlo.broadcast_compare %arg0, %arg1 {comparison_direction = #chlo<comparison_direction EQ>} :…
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/mhlo/
Dverifier_select_and_scatter_op.mlir13 comparison_direction = #mhlo<comparison_direction GE>
41 comparison_direction = #mhlo<comparison_direction GE>}
71 comparison_direction = #mhlo<comparison_direction GE>
98 comparison_direction = #mhlo<comparison_direction GE>
125 comparison_direction = #mhlo<comparison_direction GE>
153 comparison_direction = #mhlo<comparison_direction GE>
180 comparison_direction = #mhlo<comparison_direction GE>
208 comparison_direction = #mhlo<comparison_direction GE>
236 comparison_direction = #mhlo<comparison_direction GE>
262 comparison_direction = #mhlo<comparison_direction GE>
[all …]
Dlegalize-to-std.mlir63 …%0 = "mhlo.compare"(%arg0, %arg1) {comparison_direction = #mhlo<comparison_direction EQ>} : (tenso…
65 …%1 = "mhlo.compare"(%arg0, %arg1) {comparison_direction = #mhlo<comparison_direction NE>} : (tenso…
67 …%2 = "mhlo.compare"(%arg0, %arg1) {comparison_direction = #mhlo<comparison_direction LT>} : (tenso…
69 …%3 = "mhlo.compare"(%arg0, %arg1) {comparison_direction = #mhlo<comparison_direction LE>} : (tenso…
71 …%4 = "mhlo.compare"(%arg0, %arg1) {comparison_direction = #mhlo<comparison_direction GT>} : (tenso…
73 …%5 = "mhlo.compare"(%arg0, %arg1) {comparison_direction = #mhlo<comparison_direction GE>} : (tenso…
81 …%0 = "mhlo.compare"(%arg0, %arg1) {comparison_direction = #mhlo<comparison_direction EQ>} : (tenso…
83 …%1 = "mhlo.compare"(%arg0, %arg1) {comparison_direction = #mhlo<comparison_direction NE>} : (tenso…
85 …%2 = "mhlo.compare"(%arg0, %arg1) {comparison_direction = #mhlo<comparison_direction LT>} : (tenso…
87 …%3 = "mhlo.compare"(%arg0, %arg1) {comparison_direction = #mhlo<comparison_direction LE>} : (tenso…
[all …]
Dsink-constants-to-control-flow.mlir15 …%1 = "mhlo.compare"(%c0, %arg1) {comparison_direction = #mhlo<comparison_direction LT>} : (tensor<…
37 …%0 = "mhlo.compare"(%arg0, %c0) {comparison_direction = #mhlo<comparison_direction LT>} : (tensor<…
61 …%3 = "mhlo.compare"(%1, %2) {comparison_direction = #mhlo<comparison_direction GT>} : (tensor<f32>…
Dlegalize-control-flow.mlir15 …%1 = "mhlo.compare"(%arg1, %arg1) {comparison_direction = #mhlo<comparison_direction LT>, name = "…
55 …%6 = "mhlo.compare"(%arg1, %5) {comparison_direction = #mhlo<comparison_direction LT>} : (tensor<i…
94 …%0 = "mhlo.compare"(%arg0, %cst) {comparison_direction = #mhlo<comparison_direction LT>} : (tensor…
122 …%cmp1 = "mhlo.compare"(%arg0, %cst) {comparison_direction = #mhlo<comparison_direction LT>} : (ten…
126 …%cmp2 = "mhlo.compare"(%arg1, %cst) {comparison_direction = #mhlo<comparison_direction LT>} : (ten…
Dinvalid_while_op.mlir14 …%4 = "mhlo.compare"(%arg1, %3) {comparison_direction = "LT"} : (tensor<1xi32>, tensor<1xi32>) -> t…
37 …%4 = "mhlo.compare"(%arg1, %3) {comparison_direction = "LT"} : (tensor<1xi32>, tensor<1xi32>) -> t…
60 …%4 = "mhlo.compare"(%arg1, %3) {comparison_direction = "LT"} : (tensor<1xi32>, tensor<1xi32>) -> t…
83 …%4 = "mhlo.compare"(%arg1, %3) {comparison_direction = "LT"} : (tensor<1xi32>, tensor<1xi32>) -> t…
123 …%4 = "mhlo.compare"(%arg1, %3) {comparison_direction = "LT"} : (tensor<1xi32>, tensor<1xi32>) -> t…
165 …%4 = "mhlo.compare"(%arg1, %3) {comparison_direction = "LT"} : (tensor<1xi32>, tensor<1xi32>) -> t…
186 …%4 = "mhlo.compare"(%arg1, %3) {comparison_direction = "LT"} : (tensor<1xi32>, tensor<1xi32>) -> t…
209 …%4 = "mhlo.compare"(%arg1, %3) {comparison_direction = "LT"} : (tensor<1xi32>, tensor<1xi32>) -> t…
Dwhile_prettyprint.mlir26 …%1 = "mhlo.compare"(%iterArg, %iterArg_4) {comparison_direction = #mhlo<comparison_direction LT>} …
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dlegalize_hlo.mlir403 …%1 = "chlo.broadcast_compare"(%arg0, %0) {comparison_direction = #mhlo<comparison_direction LT>} :…
405 …%3 = "chlo.broadcast_compare"(%arg1, %2) {comparison_direction = #mhlo<comparison_direction LT>} :…
406 …) {broadcast_dimensions = dense<1> : tensor<1xi64>, comparison_direction = #mhlo<comparison_direct…
442 …%1 = "mhlo.compare"(%arg0, %0) {comparison_direction = #mhlo<comparison_direction LT>} : (tensor<3…
444 …%3 = "chlo.broadcast_compare"(%arg1, %2) {comparison_direction = #mhlo<comparison_direction LT>} :…
445 …) {broadcast_dimensions = dense<1> : tensor<1xi64>, comparison_direction = #mhlo<comparison_direct…
495 …%0 = "mhlo.compare"(%arg0, %arg1) {comparison_direction = #mhlo<comparison_direction EQ>} : (tenso…
506 …%0 = "chlo.broadcast_compare"(%arg0, %arg1) {comparison_direction = #mhlo<comparison_direction EQ>…
517 …) {broadcast_dimensions = dense<1> : tensor<1xi64>, comparison_direction = #mhlo<comparison_direct…
528 …) {broadcast_dimensions = dense<1> : tensor<1xi64>, comparison_direction = #mhlo<comparison_direct…
[all …]
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/
Dif.mlir17 …%0 = "mhlo.compare"(%arg0, %cst) {comparison_direction = #mhlo<comparison_direction LT>} : (tensor…
40 …%0 = "mhlo.compare"(%arg0, %cst) {comparison_direction = #mhlo<comparison_direction LT>} : (tensor…
79 …%0 = "mhlo.compare"(%arg0, %cst) {comparison_direction = #mhlo<comparison_direction LT>} : (tensor…
118 …%0 = "mhlo.compare"(%arg0, %cst) {comparison_direction = #mhlo<comparison_direction LT>} : (tensor…
158 …%0 = "mhlo.compare"(%arg0, %cst) {comparison_direction = #mhlo<comparison_direction LT>} : (tensor…
198 …%0 = "mhlo.compare"(%arg0, %cst) {comparison_direction = #mhlo<comparison_direction LT>} : (tensor…
Dwhile.mlir14 …%1 = "mhlo.compare"(%arg1, %arg1) {comparison_direction = #mhlo<comparison_direction LT>} : (tenso…
73 …%5 = "mhlo.compare"(%arg1, %arg2) {comparison_direction = #mhlo<comparison_direction LT>} : (tenso…
142 …%7 = "mhlo.compare"(%5, %6) {comparison_direction = #mhlo<comparison_direction LT>} : (tensor<i32>…
198 …%9 = "mhlo.compare"(%arg1, %arg3) {comparison_direction = #mhlo<comparison_direction LT>} : (tenso…
256 …%6 = "mhlo.compare"(%4, %5) {comparison_direction = #mhlo<comparison_direction LT>} : (tensor<f32>…
303 …%2 = "mhlo.compare"(%arg1, %arg2) {comparison_direction = #mhlo<comparison_direction LT>} : (tenso…
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tosa/tests/
Dbinary.mlir13 …%0 = "mhlo.compare"(%arg0, %arg1) {comparison_direction = #mhlo<comparison_direction EQ>} : (tenso…
20 …%0 = "mhlo.compare"(%arg0, %arg1) {comparison_direction = #mhlo<comparison_direction LT>} : (tenso…
28 …%0 = "mhlo.compare"(%arg0, %arg1) {comparison_direction = #mhlo<comparison_direction NE>} : (tenso…
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/mhlo/canonicalize/
Dcanonicalize.mlir992 …%1 = "mhlo.compare"(%arg1, %arg1) {comparison_direction = #mhlo<comparison_direction LT>} : (tenso…
1010 …%1 = "mhlo.compare"(%arg1, %arg1) {comparison_direction = #mhlo<comparison_direction LT>} : (tenso…
1064 …%0 = "mhlo.compare"(%arg0, %arg0) {comparison_direction = #mhlo<comparison_direction EQ>} : (tenso…
1071 …%0 = "mhlo.compare"(%arg0, %arg0) {comparison_direction = #mhlo<comparison_direction LE>} : (tenso…
1078 …%0 = "mhlo.compare"(%arg0, %arg0) {comparison_direction = #mhlo<comparison_direction GE>} : (tenso…
1084 …%0 = "mhlo.compare"(%arg0, %arg0) {comparison_direction = #mhlo<comparison_direction NE>} : (tenso…
1091 …%0 = "mhlo.compare"(%arg0, %arg0) {comparison_direction = #mhlo<comparison_direction LT>} : (tenso…
1098 …%0 = "mhlo.compare"(%arg0, %arg0) {comparison_direction = #mhlo<comparison_direction GT>} : (tenso…
1106 …%0 = "mhlo.compare"(%arg0, %arg0) {comparison_direction = #mhlo<comparison_direction EQ>} : (tenso…
1114 …%0 = "mhlo.compare"(%arg0, %arg0) {comparison_direction = #mhlo<comparison_direction EQ>} : (tenso…
[all …]
Dwhile.mlir16 …%1 = "mhlo.compare"(%arg5, %arg6) {comparison_direction = #mhlo<comparison_direction LT>} : (tenso…
44 …%1 = "mhlo.compare"(%arg0, %arg0) {comparison_direction = #mhlo<comparison_direction LT>} : (tenso…
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dprepare-for-export.mlir33 …%1 = "mhlo.compare"(%arg0, %arg1) {comparison_direction = #mhlo<comparison_direction LT>} : (tenso…
79 …%1 = "mhlo.compare"(%arg0, %arg1) {comparison_direction = #mhlo<comparison_direction LT>} : (tenso…
Dlegalize-tf-control-flow.mlir7 …%0 = "mhlo.compare"(%arg0, %arg1) {comparison_direction = #mhlo<comparison_direction GT>} : (tenso…
37 …%0 = "mhlo.compare"(%arg0, %arg1) {comparison_direction = #mhlo<comparison_direction GT>} : (tenso…
156 …%4 = "mhlo.compare"(%carg2, %3) {comparison_direction = #mhlo<comparison_direction LT>} : (tensor<…
212 …%2 = "mhlo.compare"(%0, %1) {comparison_direction = #mhlo<comparison_direction LT>} : (tensor<i32>…
Dlegalize-tf-no-tf2xla-fallback.mlir600 …25:.*]] = chlo.broadcast_compare %[[V18]], %[[V5]] {comparison_direction = #mhlo<comparison_direct…
609 …34:.*]] = chlo.broadcast_compare %[[V32]], %[[V5]] {comparison_direction = #mhlo<comparison_direct…
610 …5:.*]] = chlo.broadcast_compare %[[V32]], %[[V15]] {comparison_direction = #mhlo<comparison_direct…
612 …37:.*]] = chlo.broadcast_compare %[[V33]], %[[V5]] {comparison_direction = #mhlo<comparison_direct…
613 …8:.*]] = chlo.broadcast_compare %[[V33]], %[[V13]] {comparison_direction = #mhlo<comparison_direct…
669 … = chlo.broadcast_compare %{{[0-9]*}}, %{{[0-9]*}} {comparison_direction = #mhlo<comparison_direct…
683 … = chlo.broadcast_compare %{{[0-9]*}}, %{{[0-9]*}} {comparison_direction = #mhlo<comparison_direct…
774 …CMP1:%.+]] = chlo.broadcast_compare [[MUL]], %arg0 {comparison_direction = #mhlo<comparison_direct…
776 …2:%.+]] = chlo.broadcast_compare %arg0, [[ZEROS1]] {comparison_direction = #mhlo<comparison_direct…
778 …3:%.+]] = chlo.broadcast_compare %arg1, [[ZEROS2]] {comparison_direction = #mhlo<comparison_direct…
[all …]
Dlegalize-tf.mlir607 …25:.*]] = chlo.broadcast_compare %[[V18]], %[[V5]] {comparison_direction = #mhlo<comparison_direct…
616 …34:.*]] = chlo.broadcast_compare %[[V32]], %[[V5]] {comparison_direction = #mhlo<comparison_direct…
617 …5:.*]] = chlo.broadcast_compare %[[V32]], %[[V15]] {comparison_direction = #mhlo<comparison_direct…
619 …37:.*]] = chlo.broadcast_compare %[[V33]], %[[V5]] {comparison_direction = #mhlo<comparison_direct…
620 …8:.*]] = chlo.broadcast_compare %[[V33]], %[[V13]] {comparison_direction = #mhlo<comparison_direct…
676 … = chlo.broadcast_compare %{{[0-9]*}}, %{{[0-9]*}} {comparison_direction = #mhlo<comparison_direct…
690 … = chlo.broadcast_compare %{{[0-9]*}}, %{{[0-9]*}} {comparison_direction = #mhlo<comparison_direct…
781 …CMP1:%.+]] = chlo.broadcast_compare [[MUL]], %arg0 {comparison_direction = #mhlo<comparison_direct…
783 …2:%.+]] = chlo.broadcast_compare %arg0, [[ZEROS1]] {comparison_direction = #mhlo<comparison_direct…
785 …3:%.+]] = chlo.broadcast_compare %arg1, [[ZEROS2]] {comparison_direction = #mhlo<comparison_direct…
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dlower_upper_bound_ops.cc34 xla::ComparisonDirection comparison_direction) { in BuildLowerUpperBoundOp() argument
65 comparison_direction); in BuildLowerUpperBoundOp()
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/
Drank-specialization.mlir127 ….*]] = chlo.broadcast_compare %[[ARG1]], %[[ZERO]] {comparison_direction = #mhlo<comparison_direct…
132 %1 = chlo.broadcast_compare %arg0, %0 {comparison_direction = #mhlo<comparison_direction GT>}
342 ….*]] = chlo.broadcast_compare %[[ARG0_]], %[[C0_]] {comparison_direction = #mhlo<comparison_direct…
350 %2 = chlo.broadcast_compare %arg0, %1 {comparison_direction = #mhlo<comparison_direction EQ>}
375 …]] = chlo.broadcast_compare %[[SCALAR]], %[[ZERO]] {comparison_direction = #mhlo<comparison_direct…
390 ….broadcast_compare %[[FLAT_NON_SCALAR]], %[[ZERO]] {comparison_direction = #mhlo<comparison_direct…
405 …= chlo.broadcast_compare %[[FLAT_ARG0]], %[[ZERO]] {comparison_direction = #mhlo<comparison_direct…
427 …hlo.broadcast_compare %[[REDUCED_ARG0]], %[[ZERO]] {comparison_direction = #mhlo<comparison_direct…
631 …%1 = "mhlo.compare"(%arg0, %0) {comparison_direction = #mhlo<comparison_direction GT>} : (tensor<*…
675 %5 = "mhlo.compare"(%arg0, %4) {comparison_direction = #mhlo<comparison_direction GT>}
/external/tensorflow/tensorflow/compiler/mlir/hlo/tests/
Dmhlo_flatten_tuple.mlir11 // CHECK: %[[COMPARE_0:.*]] = "mhlo.compare"(%arg1, %[[SLICE_0]]) {comparison_direction …
29 …%4 = "mhlo.compare"(%arg1, %3) {comparison_direction = "LT"} : (tensor<1xi32>, tensor<1xi32>) -> t…
47 // CHECK: %[[COMPARE_0:.*]] = "mhlo.compare"(%arg1, %[[SLICE_0]]) {comparison_direction …
68 …%4 = "mhlo.compare"(%t0, %3) {comparison_direction = "LT"} : (tensor<1xi32>, tensor<1xi32>) -> ten…
92 // CHECK: %[[COMPARE_0:.*]] = "mhlo.compare"(%arg1, %[[SLICE_0]]) {comparison_direction …
116 …%5 = "mhlo.compare"(%t0, %4) {comparison_direction = "LT"} : (tensor<1xi32>, tensor<1xi32>) -> ten…
142 // CHECK: %[[COMPARE_0:.*]] = "mhlo.compare"(%arg1, %[[SLICE_0]]) {comparison_direction …
164 …%5 = "mhlo.compare"(%arg1, %4) {comparison_direction = "LT"} : (tensor<1xi32>, tensor<1xi32>) -> t…
195 // CHECK: %[[COMPARE_0:.*]] = "mhlo.compare"(%arg1, %arg3) {comparison_direction = "LT"}…
214 …%4 = "mhlo.compare"(%t0, %t2) {comparison_direction = "LT"} : (tensor<i32>, tensor<i32>) -> tensor…
[all …]
Drank-specialization.mlir127 …// CHECK: %[[CMP_GT:.*]] = chlo.broadcast_compare %[[ARG1]], %[[ZERO]] {comparison_direction = "…
132 %1 = chlo.broadcast_compare %arg0, %0 {comparison_direction = "GT"}
342 …// CHECK: %[[TMP0:.*]] = chlo.broadcast_compare %[[ARG0_]], %[[C0_]] {comparison_direction = "EQ…
350 %2 = chlo.broadcast_compare %arg0, %1 {comparison_direction = "EQ"}
375 … %[[PRED:.*]] = chlo.broadcast_compare %[[SCALAR]], %[[ZERO]] {comparison_direction = "EQ"} : (…
390 …ED:.*]] = chlo.broadcast_compare %[[FLAT_NON_SCALAR]], %[[ZERO]] {comparison_direction = "EQ"} : (…
405 … %[[PRED:.*]] = chlo.broadcast_compare %[[FLAT_ARG0]], %[[ZERO]] {comparison_direction = "EQ"} : (…
427 …[PRED:.*]] = chlo.broadcast_compare %[[REDUCED_ARG0]], %[[ZERO]] {comparison_direction = "EQ"} : (…
626 // CHECK: %[[TMP1:.*]] = "mhlo.compare"(%[[ARG0_]], %[[TMP0]]) {comparison_direction = "GT"}
631 …%1 = "mhlo.compare"(%arg0, %0) {comparison_direction = "GT"} : (tensor<*xf32>, tensor<*xf32>) -> t…
[all …]
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tests/Dialect/chlo/
Dchlo_infer_shape_type_methods.mlir30 …%0 = chlo.broadcast_compare %arg0, %arg1 {comparison_direction = #mhlo<comparison_direction EQ>} :…

123