| /external/tensorflow/tensorflow/compiler/xla/ |
| D | comparison_util.cc | 22 std::string ComparisonDirectionToString(ComparisonDirection direction) { in ComparisonDirectionToString()
24 case ComparisonDirection::kEq: in ComparisonDirectionToString()
26 case ComparisonDirection::kNe: in ComparisonDirectionToString()
28 case ComparisonDirection::kGe: in ComparisonDirectionToString()
30 case ComparisonDirection::kGt: in ComparisonDirectionToString()
32 case ComparisonDirection::kLe: in ComparisonDirectionToString()
34 case ComparisonDirection::kLt: in ComparisonDirectionToString()
39 StatusOr<ComparisonDirection> StringToComparisonDirection( in StringToComparisonDirection()
42 new absl::flat_hash_map<string, ComparisonDirection>({ in StringToComparisonDirection()
43 {"EQ", ComparisonDirection::kEq}, in StringToComparisonDirection()
[all …]
|
| D | comparison_util.h | 26 enum class ComparisonDirection : uint8 { enum
35 string ComparisonDirectionToString(ComparisonDirection direction);
37 StatusOr<ComparisonDirection> StringToComparisonDirection(
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_matchers.h | 61 ComparisonDirection direction, in HloComparisonMatcher()
69 ComparisonDirection direction_;
273 ComparisonDirection::kEq, {operands...})); in Eq()
278 ComparisonDirection::kNe, {operands...})); in Ne()
283 ComparisonDirection::kGe, {operands...})); in Ge()
288 ComparisonDirection::kGt, {operands...})); in Gt()
293 ComparisonDirection::kLe, {operands...})); in Le()
298 ComparisonDirection::kLt, {operands...})); in Lt()
|
| D | hlo_evaluator.cc | 59 StatusOr<Literal> Compare(const Shape& shape, ComparisonDirection direction, in Compare()
63 case ComparisonDirection::kEq: in Compare()
68 case ComparisonDirection::kNe: in Compare()
73 case ComparisonDirection::kGe: in Compare()
78 case ComparisonDirection::kGt: in Compare()
83 case ComparisonDirection::kLe: in Compare()
88 case ComparisonDirection::kLt: in Compare()
107 ComparisonDirection direction, in Compare()
112 case ComparisonDirection::kEq: in Compare()
117 case ComparisonDirection::kNe: in Compare()
[all …]
|
| D | hlo_graph_dumper.cc | 264 case ComparisonDirection::kLe: in MatchTrivialComputation()
265 case ComparisonDirection::kGe: in MatchTrivialComputation()
266 case ComparisonDirection::kGt: in MatchTrivialComputation()
267 case ComparisonDirection::kLt: in MatchTrivialComputation()
287 case ComparisonDirection::kLe: in MatchTrivialComputation()
289 case ComparisonDirection::kGe: in MatchTrivialComputation()
291 case ComparisonDirection::kGt: in MatchTrivialComputation()
293 case ComparisonDirection::kLt: in MatchTrivialComputation()
295 case ComparisonDirection::kEq: in MatchTrivialComputation()
297 case ComparisonDirection::kNe: in MatchTrivialComputation()
|
| D | hlo_matchers_test.cc | 228 ComparisonDirection::kEq); in TEST()
230 ComparisonDirection::kNe); in TEST()
234 ComparisonDirection::kLe); in TEST()
|
| D | stable_sort_expander.cc | 184 scalar_pred, old_root, cloned_root, ComparisonDirection::kEq)); in ExpandInstruction()
189 ComparisonDirection::kLt)); in ExpandInstruction()
|
| D | flatten_call_graph_test.cc | 86 zero, ComparisonDirection::kGt)); in MakeConditionComputation()
164 ComparisonDirection::kEq)); in TEST_F()
|
| D | elemental_ir_emitter.cc | 723 case ComparisonDirection::kEq: in EmitFloatBinaryOp()
726 case ComparisonDirection::kNe: in EmitFloatBinaryOp()
729 case ComparisonDirection::kLt: in EmitFloatBinaryOp()
732 case ComparisonDirection::kGt: in EmitFloatBinaryOp()
735 case ComparisonDirection::kLe: in EmitFloatBinaryOp()
738 case ComparisonDirection::kGe: in EmitFloatBinaryOp()
846 case ComparisonDirection::kEq: in EmitComplexBinaryOp()
853 case ComparisonDirection::kNe: in EmitComplexBinaryOp()
1292 case ComparisonDirection::kEq: in EmitIntegerBinaryOp()
1295 case ComparisonDirection::kNe: in EmitIntegerBinaryOp()
[all …]
|
| D | scatter_expander.cc | 186 MakeCompareHlo(ComparisonDirection::kLe, zero_index, index)); in CheckIndexValidity()
198 MakeCompareHlo(ComparisonDirection::kGe, in CheckIndexValidity()
|
| D | while_loop_analysis.cc | 289 .WithComparisonDirection(ComparisonDirection::kLt) in PatternMatchLoopTripCount()
306 .WithComparisonDirection(ComparisonDirection::kLe) in PatternMatchLoopTripCount()
|
| D | pattern_matcher.h | 1678 ComparisonDirection direction)
1706 ComparisonDirection direction_;
1996 auto WithComparisonDirection(ComparisonDirection direction) const
2287 .WithComparisonDirection(ComparisonDirection::k##NAME)) { \
2290 .WithComparisonDirection(ComparisonDirection::k##NAME); \
2298 .WithComparisonDirection(ComparisonDirection::k##NAME)) { \
2303 .WithComparisonDirection(ComparisonDirection::k##NAME); \
2312 .WithComparisonDirection(ComparisonDirection::k##NAME)) { \
2317 .WithComparisonDirection(ComparisonDirection::k##NAME); \
|
| D | bfloat16_propagation_test.cc | 113 ShapeUtil::MakeShape(PRED, {2, 4}), a, b, ComparisonDirection::kEq)); in TEST_F()
589 ComparisonDirection::kGt)); in TEST_F()
647 ComparisonDirection::kGt)); in TEST_F()
722 ComparisonDirection::kGt)); in TEST_F()
818 ComparisonDirection::kGt)); in TEST_F()
847 ComparisonDirection::kGt)); in TEST_F()
|
| D | dynamic_padder.cc | 166 ComparisonDirection::kLt)); in Run()
|
| D | hlo_creation_utils.h | 37 StatusOr<HloInstruction*> MakeCompareHlo(ComparisonDirection direction,
|
| D | stable_sort_expander_test.cc | 69 ASSERT_EQ(root->operand(0)->comparison_direction(), ComparisonDirection::kEq); in CheckComputationHasTieBreaker()
|
| D | pattern_matcher_test.cc | 939 ComparisonDirection::kEq); in TEST()
941 ComparisonDirection::kNe); in TEST()
945 ComparisonDirection::kLe); in TEST()
|
| D | hlo_instruction.cc | 69 absl::optional<ComparisonDirection> comparison_direction; in CreateFromProto()
77 comparison_direction = ComparisonDirection::kEq; in CreateFromProto()
79 comparison_direction = ComparisonDirection::kNe; in CreateFromProto()
81 comparison_direction = ComparisonDirection::kGe; in CreateFromProto()
83 comparison_direction = ComparisonDirection::kGt; in CreateFromProto()
85 comparison_direction = ComparisonDirection::kLe; in CreateFromProto()
87 comparison_direction = ComparisonDirection::kLt; in CreateFromProto()
799 ComparisonDirection direction) { in CreateCompare()
3541 ComparisonDirection HloInstruction::comparison_direction() const { in comparison_direction()
|
| D | while_util.cc | 169 MakeCompareHlo(ComparisonDirection::kLt, indvar, trip_count_constant)); in MakeCountedLoopConditionComputation()
|
| D | hlo_memory_scheduler_test.cc | 259 zero_vector, ComparisonDirection::kNe)); in TEST_F()
|
| D | hlo_instructions.h | 138 ComparisonDirection direction);
139 ComparisonDirection direction() const { return direction_; } in direction()
153 ComparisonDirection direction_;
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | fusion_test.cc | 68 absl::optional<ComparisonDirection> direction = absl::nullopt) { in TestElementwise2D()
142 bool ComputeElementwiseAnswerCompare(ComparisonDirection direction,
175 bool FusionTest::ComputeElementwiseAnswerCompare(ComparisonDirection direction, in ComputeElementwiseAnswerCompare()
178 case ComparisonDirection::kEq: in ComputeElementwiseAnswerCompare()
180 case ComparisonDirection::kNe: in ComputeElementwiseAnswerCompare()
182 case ComparisonDirection::kGt: in ComputeElementwiseAnswerCompare()
184 case ComparisonDirection::kLt: in ComputeElementwiseAnswerCompare()
186 case ComparisonDirection::kGe: in ComputeElementwiseAnswerCompare()
188 case ComparisonDirection::kLe: in ComputeElementwiseAnswerCompare()
760 TestElementwise2D<bool, 2>(HloOpcode::kCompare, ComparisonDirection::kEq); in XLA_TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_index_test.cc | 53 ComparisonDirection::kGe)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | while_transformer_test.cc | 49 ComparisonDirection::kLt)); in BuildConditionComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 484 absl::optional<ComparisonDirection> direction) { in BinaryOp()
2923 return Compare(lhs, rhs, broadcast_dimensions, ComparisonDirection::kEq); in Eq()
2928 return Compare(lhs, rhs, broadcast_dimensions, ComparisonDirection::kNe); in Ne()
2933 return Compare(lhs, rhs, broadcast_dimensions, ComparisonDirection::kGe); in Ge()
2938 return Compare(lhs, rhs, broadcast_dimensions, ComparisonDirection::kGt); in Gt()
2943 return Compare(lhs, rhs, broadcast_dimensions, ComparisonDirection::kLe); in Le()
2948 return Compare(lhs, rhs, broadcast_dimensions, ComparisonDirection::kLt); in Lt()
2953 ComparisonDirection direction) { in Compare()
|