| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | comparison_expander.cc | 86 PrimitiveType compare_type = compare_shape.element_type(); in ExpandInstruction() local
87 CHECK(primitive_util::IsFloatingPointType(compare_type)); in ExpandInstruction()
91 if (compare_type == BF16) { in ExpandInstruction()
92 compare_type = F32; in ExpandInstruction()
93 compare_shape.set_element_type(compare_type); in ExpandInstruction()
100 int64 bit_width = primitive_util::BitWidth(compare_type); in ExpandInstruction()
|
| /external/pcre/dist2/src/ |
| D | pcre2_jit_simd_inc.h | 106 …ast_forward_char_pair_sse2_compare(struct sljit_compiler *compiler, sse2_compare_type compare_type, in fast_forward_char_pair_sse2_compare() argument
115 if (compare_type != sse2_compare_match2) in fast_forward_char_pair_sse2_compare()
119 if (compare_type == sse2_compare_match1i) in fast_forward_char_pair_sse2_compare()
194 sse2_compare_type compare_type = sse2_compare_match1; in fast_forward_char_simd() local
210 compare_type = sse2_compare_match1i; in fast_forward_char_simd()
215 compare_type = sse2_compare_match2; in fast_forward_char_simd()
270 …fast_forward_char_pair_sse2_compare(compiler, compare_type, i, data_ind, cmp1_ind, cmp2_ind, tmp_i… in fast_forward_char_simd()
297 …fast_forward_char_pair_sse2_compare(compiler, compare_type, i, data_ind, cmp1_ind, cmp2_ind, tmp_i… in fast_forward_char_simd()
355 sse2_compare_type compare_type = sse2_compare_match1; in fast_requested_char_simd() local
369 compare_type = sse2_compare_match1i; in fast_requested_char_simd()
[all …]
|
| D | pcre2_jit_neon_inc.h | 114 compare_type compare1_type = compare_match1; in FF_FUN()
115 compare_type compare2_type = compare_match1; in FF_FUN()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | chlo_ops.cc | 221 StringAttr compare_type) { in build() argument
225 comparison_direction, compare_type); in build()
|
| D | hlo_ops.cc | 2835 StringAttr compare_type) { in build() argument
2839 compare_type); in build()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | chlo_ops.td | 658 and `compare_type`. If unspecified, `compare_type` is FLOAT for float element
671 OptionalAttr<HLO_ComparisonTypeAttr>:$compare_type
678 "StringAttr":$comparison_direction, CArg<"StringAttr", "{}">:$compare_type)>];
|
| D | hlo_ops.td | 686 OptionalAttr<HLO_ComparisonTypeAttr>:$compare_type
694 "StringAttr":$comparison_direction, CArg<"StringAttr", "{}">:$compare_type)>,
|
| D | hlo_ops_base.td | 732 and `compare_type`. If unspecified, `compare_type` is FLOAT for float element
|
| D | lhlo_ops.td | 292 OptionalAttr<HLO_ComparisonTypeAttr>:$compare_type
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | export.mlir | 804 …%2 = "mhlo.compare"(%arg3, %arg4) {compare_type = "TOTALORDER", comparison_direction = "GE"} : (te…
1004 …%7 = "mhlo.compare"(%arg0, %arg1) {compare_type = "FLOAT", comparison_direction = "GT"} : (tensor<…
1023 …%7 = "mhlo.compare"(%arg0, %arg1) {compare_type = "FLOAT", comparison_direction = "GT"} : (tensor<…
|
| D | import.hlotxt | 151 …// CHECK-NEXT: "mhlo.compare"(%arg0, %arg1) {compare_type = "TOTALORDER", comparison_direction = …
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 994 auto compare_type = [&]() { in EmitCompareOp() local
1007 builder_.getStringAttr(stringifyComparisonType(compare_type))); in EmitCompareOp()
|
| D | legalize_tf.cc | 1049 llvm::Optional<StringRef> compare_type, in BuildSortComparisonBody() argument
1063 if (compare_type) type_attr = builder->getStringAttr(*compare_type); in BuildSortComparisonBody()
4763 llvm::Optional<StringRef> compare_type; in matchAndRewrite() local
4765 compare_type.emplace("TOTALORDER"); in matchAndRewrite()
4767 /*direction=*/"GT", compare_type, in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf-with-tf2xla.mlir | 125 …// CHECK-NEXT: "mhlo.compare"(%arg0, %arg1) {compare_type = "SIGNED", comparison_direction = "GT"}
|
| D | legalize-tf.mlir | 3954 …// CHECK-NEXT: %[[CMP:.*]] = "mhlo.compare"(%[[LHS]], %[[RHS]]) {compare_type = "TOTALORDER", co…
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 4210 auto compare_type = in CompareTotalOrder() local
4215 compare_type); in CompareTotalOrder()
4289 ComparisonDirection direction, Comparison::Type compare_type) { in Compare() argument
4291 broadcast_dimensions, direction, compare_type); in Compare()
|
| D | xla_builder.h | 1118 Comparison::Type compare_type);
1850 ComparisonDirection direction, Comparison::Type compare_type);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | legalize_hlo.mlir | 546 …pare"(%arg0, %arg1) {broadcast_dimensions = dense<1> : tensor<1xi64>, compare_type = "TOTALORDER",…
619 …%0 = "mhlo.compare"(%arg0, %arg1) {compare_type = "TOTALORDER", comparison_direction = "GT"} : (te…
|