Home
last modified time | relevance | path

Searched refs:ComparisonDirection (Results 1 – 25 of 89) sorted by relevance

1234

/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_standard.cc58 case ComparisonDirection::EQ: in matchAndRewrite()
61 case ComparisonDirection::NE: in matchAndRewrite()
64 case ComparisonDirection::LT: in matchAndRewrite()
67 case ComparisonDirection::LE: in matchAndRewrite()
70 case ComparisonDirection::GT: in matchAndRewrite()
73 case ComparisonDirection::GE: in matchAndRewrite()
106 case ComparisonDirection::EQ: in matchAndRewrite()
109 case ComparisonDirection::NE: in matchAndRewrite()
112 case ComparisonDirection::LT: in matchAndRewrite()
115 case ComparisonDirection::LE: in matchAndRewrite()
[all …]
Dchlo_legalize_to_hlo.cc127 mhlo::ComparisonDirection::LE); in materializeBesselI1eApproximation()
332 loc, absX, eight, mhlo::ComparisonDirection::LT); in materializeErfcApproximationF64ForMagnituteGeOne()
340 mhlo::ComparisonDirection::LT); in materializeErfcApproximationF64ForMagnituteGeOne()
350 loc, x, zero, mhlo::ComparisonDirection::LT); in materializeErfcApproximationF64ForMagnituteGeOne()
408 loc, absX, one, mhlo::ComparisonDirection::LT); in materializeErfApproximationF64()
434 loc, absX, one, mhlo::ComparisonDirection::LT); in materializeErfcApproximationF64()
476 loc, absX, two, mhlo::ComparisonDirection::LT); in materializeErfcApproximationF32ForMagnitudeGeOne()
488 mhlo::ComparisonDirection::LT); in materializeErfcApproximationF32ForMagnitudeGeOne()
498 loc, x, zero, mhlo::ComparisonDirection::LT); in materializeErfcApproximationF32ForMagnitudeGeOne()
579 loc, absX, one, mhlo::ComparisonDirection::LT); in materializeErfcApproximationF32()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dlower_upper_bound_ops.cc34 xla::ComparisonDirection comparison_direction) { in BuildLowerUpperBoundOp()
90 BuildLowerUpperBoundOp(ctx, out_dtype_, xla::ComparisonDirection::kGt); in Compile()
106 BuildLowerUpperBoundOp(ctx, out_dtype_, xla::ComparisonDirection::kGe); in Compile()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcpu_gpu_fusion_test.cc65 std::optional<ComparisonDirection> direction = std::nullopt) { in TestElementwise2D()
139 bool ComputeElementwiseAnswerCompare(ComparisonDirection direction,
173 ComparisonDirection direction, absl::Span<const float> xs) { in ComputeElementwiseAnswerCompare()
175 case ComparisonDirection::kEq: in ComputeElementwiseAnswerCompare()
177 case ComparisonDirection::kNe: in ComputeElementwiseAnswerCompare()
179 case ComparisonDirection::kGt: in ComputeElementwiseAnswerCompare()
181 case ComparisonDirection::kLt: in ComputeElementwiseAnswerCompare()
183 case ComparisonDirection::kGe: in ComputeElementwiseAnswerCompare()
185 case ComparisonDirection::kLe: in ComputeElementwiseAnswerCompare()
758 TestElementwise2D<bool, 2>(HloOpcode::kCompare, ComparisonDirection::kEq); in XLA_TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_matchers.h64 ComparisonDirection direction, in HloComparisonMatcher()
72 ComparisonDirection direction_;
355 ComparisonDirection::kEq, {operands...})); in Eq()
360 ComparisonDirection::kNe, {operands...})); in Ne()
365 ComparisonDirection::kGe, {operands...})); in Ge()
370 ComparisonDirection::kGt, {operands...})); in Gt()
375 ComparisonDirection::kLe, {operands...})); in Le()
380 ComparisonDirection::kLt, {operands...})); in Lt()
Dstable_sort_expander.cc184 scalar_pred, old_root, cloned_root, ComparisonDirection::kEq)); in ExpandInstruction()
189 ComparisonDirection::kLt)); in ExpandInstruction()
Dhlo_graph_dumper.cc279 case ComparisonDirection::kLe: in MatchTrivialComputation()
280 case ComparisonDirection::kGe: in MatchTrivialComputation()
281 case ComparisonDirection::kGt: in MatchTrivialComputation()
282 case ComparisonDirection::kLt: in MatchTrivialComputation()
308 case ComparisonDirection::kLe: in MatchTrivialComputation()
310 case ComparisonDirection::kGe: in MatchTrivialComputation()
312 case ComparisonDirection::kGt: in MatchTrivialComputation()
314 case ComparisonDirection::kLt: in MatchTrivialComputation()
316 case ComparisonDirection::kEq: in MatchTrivialComputation()
318 case ComparisonDirection::kNe: in MatchTrivialComputation()
Dhlo_matchers_test.cc244 ComparisonDirection::kEq); in TEST_F()
246 ComparisonDirection::kNe); in TEST_F()
250 ComparisonDirection::kLe); in TEST_F()
Dflatten_call_graph_test.cc86 zero, ComparisonDirection::kGt)); in MakeConditionComputation()
164 ComparisonDirection::kEq)); in TEST_F()
Ddynamic_padder.cc83 .WithComparisonDirection(ComparisonDirection::kGe))) { in ChooseIdentityValue()
304 pred_shape, iota, broadcasted_effective_size, ComparisonDirection::kLt)); in PadWithScalar()
386 ComparisonDirection::kLt)); in GenerateBinaryMask()
700 rhs_key, ComparisonDirection::kGt)); in RewriteDynamicReshapeCombineInput()
1369 dynamic_size_broadcasted, ComparisonDirection::kLt)); in RewriteDynamicSort()
1508 dim_1, ComparisonDirection::kLt), in RewriteDynamicBinaryOp()
1512 one, ComparisonDirection::kEq), in RewriteDynamicBinaryOp()
1523 dim_0, ComparisonDirection::kLt), in RewriteDynamicBinaryOp()
1527 one, ComparisonDirection::kEq), in RewriteDynamicBinaryOp()
1635 mask_shape_pred, iota, broadcast_dim, ComparisonDirection::kLt)); in RewriteDynamicUpdateSlice()
Dhlo_evaluator.cc74 StatusOr<Literal> Compare(const Shape& shape, ComparisonDirection direction, in Compare()
78 case ComparisonDirection::kEq: in Compare()
83 case ComparisonDirection::kNe: in Compare()
88 case ComparisonDirection::kGe: in Compare()
93 case ComparisonDirection::kGt: in Compare()
98 case ComparisonDirection::kLe: in Compare()
103 case ComparisonDirection::kLt: in Compare()
122 ComparisonDirection direction, in Compare()
127 case ComparisonDirection::kEq: in Compare()
132 case ComparisonDirection::kNe: in Compare()
[all …]
Dcomparison_expander.cc57 compare_shape, signed_value, zero, ComparisonDirection::kLt)); in BitcastConvertFloatingPointToIntegral()
Dbfloat16_propagation_test.cc114 ShapeUtil::MakeShape(PRED, {2, 4}), a, b, ComparisonDirection::kEq)); in TEST_F()
576 ComparisonDirection::kGt)); in TEST_F()
634 ComparisonDirection::kGt)); in TEST_F()
709 ComparisonDirection::kGt)); in TEST_F()
805 ComparisonDirection::kGt)); in TEST_F()
834 ComparisonDirection::kGt)); in TEST_F()
Dalgebraic_simplifier.cc975 .WithComparisonDirection(ComparisonDirection::kLt))) { in TrySimplifyTautologicalCompare()
981 .WithComparisonDirection(ComparisonDirection::kGt))) { in TrySimplifyTautologicalCompare()
995 ComparisonDirection::kLt))); in TrySimplifyTautologicalCompare()
1698 changed_shape, a, zero_like_a, ComparisonDirection::kLt)); in TryDivideToShift()
2833 ComparisonDirection::kGe)); in HandleGather()
2877 ComparisonDirection::kLt); in MinMaxToClamp()
3585 if (compare->comparison_direction() == ComparisonDirection::kLt && in HandleCompare()
3590 if (compare->comparison_direction() == ComparisonDirection::kGe && in HandleCompare()
3595 if (compare->comparison_direction() == ComparisonDirection::kGt && in HandleCompare()
3600 if (compare->comparison_direction() == ComparisonDirection::kLe && in HandleCompare()
[all …]
Dscatter_expander.cc188 MakeCompareHlo(ComparisonDirection::kLe, zero_index, index)); in CheckIndexValidity()
200 MakeCompareHlo(ComparisonDirection::kGe, in CheckIndexValidity()
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/stablehlo/dialect/
DChloEnums.td27 // These mirror the XLA ComparisonDirection enum.
35 def CHLO_ComparisonDirection : I32EnumAttr<"ComparisonDirection",
DStablehloEnums.td102 // These mirror the XLA ComparisonDirection enum.
110 def StableHLO_ComparisonDirection : I32EnumAttr<"ComparisonDirection",
DStablehloOps.h198 bool isStable, ComparisonDirection direction);
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/tosa/lib/Transforms/
Dlegalize_mhlo.cc54 case mlir::mhlo::ComparisonDirection::EQ: { in matchAndRewrite()
59 case mlir::mhlo::ComparisonDirection::NE: { in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_ops.h121 bool isStable, ComparisonDirection direction);
Dhlo_ops_base_enums.td104 // These mirror the XLA ComparisonDirection enum.
112 def HLO_ComparisonDirection : I32EnumAttr<"ComparisonDirection",
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_index_test.cc52 ComparisonDirection::kGe)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/transforms/
Dmap_mhlo_to_scalar_op.h354 inline Optional<PredicateType> getCmpPredicate(mhlo::ComparisonDirection,
361 mhlo::ComparisonDirection comparisonDirection, bool isSigned) {
376 mhlo::ComparisonDirection comparisonDirection, bool isSigned) {
393 ComparisonDirection comparisonDirection,
418 if (comparisonDirection == ComparisonDirection::EQ) {
421 if (comparisonDirection == ComparisonDirection::NE) {
776 ComparisonDirection /*comparison_direction*/,
789 static Value map(Location loc, ComparisonDirection comparisonDirection,
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dwhile_transformer_test.cc49 ComparisonDirection::kLt)); in BuildConditionComputation()
Dcusolver_rewriter.cc114 info, zeros, ComparisonDirection::kEq)); in CreateCholesky()

1234