Searched refs:kEq (Results 1 – 18 of 18) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | comparison_util.cc | 24 case ComparisonDirection::kEq: in ComparisonDirectionToString()
43 {"EQ", ComparisonDirection::kEq}, in StringToComparisonDirection()
|
| D | comparison_util.h | 27 kEq, enumerator
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | stable_sort_expander.cc | 184 scalar_pred, old_root, cloned_root, ComparisonDirection::kEq)); in ExpandInstruction()
|
| D | stable_sort_expander_test.cc | 69 ASSERT_EQ(root->operand(0)->comparison_direction(), ComparisonDirection::kEq); in CheckComputationHasTieBreaker()
|
| D | flatten_call_graph_test.cc | 164 ComparisonDirection::kEq)); in TEST_F()
|
| D | hlo_matchers_test.cc | 228 ComparisonDirection::kEq); in TEST()
|
| D | hlo_matchers.h | 273 ComparisonDirection::kEq, {operands...})); in Eq()
|
| D | convolution_group_converter.cc | 196 ComparisonDirection::kEq)); in GetExpandedFilterMask()
|
| D | hlo_evaluator.cc | 63 case ComparisonDirection::kEq: in Compare()
112 case ComparisonDirection::kEq: in Compare()
144 case ComparisonDirection::kEq: in Compare()
|
| D | tuple_points_to_analysis_test.cc | 938 out_shape, param0, param1, ComparisonDirection::kEq)); in TEST_F()
1189 ShapeUtil::MakeShape(PRED, {}), data, data, ComparisonDirection::kEq)); in TEST_F()
|
| D | elemental_ir_emitter.cc | 723 case ComparisonDirection::kEq: in EmitFloatBinaryOp()
846 case ComparisonDirection::kEq: in EmitComplexBinaryOp()
1292 case ComparisonDirection::kEq: in EmitIntegerBinaryOp()
|
| D | pattern_matcher_test.cc | 939 ComparisonDirection::kEq); in TEST()
|
| D | hlo_graph_dumper.cc | 295 case ComparisonDirection::kEq: in MatchTrivialComputation()
|
| D | bfloat16_propagation_test.cc | 113 ShapeUtil::MakeShape(PRED, {2, 4}), a, b, ComparisonDirection::kEq)); in TEST_F()
|
| D | hlo_dataflow_analysis_test.cc | 2271 out_shape, param0, param1, ComparisonDirection::kEq)); in TEST_F()
2595 ShapeUtil::MakeShape(PRED, {}), data, data, ComparisonDirection::kEq)); in TEST_F()
|
| D | hlo_instruction.cc | 77 comparison_direction = ComparisonDirection::kEq; in CreateFromProto()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | fusion_test.cc | 178 case ComparisonDirection::kEq: in ComputeElementwiseAnswerCompare()
760 TestElementwise2D<bool, 2>(HloOpcode::kCompare, ComparisonDirection::kEq); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2923 return Compare(lhs, rhs, broadcast_dimensions, ComparisonDirection::kEq); in Eq()
|