Home
last modified time | relevance | path

Searched refs:CreateCompare (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_matchers_test.cc227 auto eq = HloInstruction::CreateCompare(shape, p0.get(), p1.get(), in TEST()
229 auto ne = HloInstruction::CreateCompare(shape, p0.get(), p1.get(), in TEST()
233 auto le = HloInstruction::CreateCompare(shape, p0.get(), add.get(), in TEST()
Dstable_sort_expander.cc183 comparator->AddInstruction(HloInstruction::CreateCompare( in ExpandInstruction()
186 comparator->AddInstruction(HloInstruction::CreateCompare( in ExpandInstruction()
Dflatten_call_graph_test.cc85 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), param0, in MakeConditionComputation()
162 builder.AddInstruction(HloInstruction::CreateCompare( in TEST_F()
Dbfloat16_propagation_test.cc112 HloInstruction* pred = builder.AddInstruction(HloInstruction::CreateCompare( in TEST_F()
577 auto cond_root = builder_cond.AddInstruction(HloInstruction::CreateCompare( in TEST_F()
635 builder_cond.AddInstruction(HloInstruction::CreateCompare( in TEST_F()
710 builder_cond.AddInstruction(HloInstruction::CreateCompare( in TEST_F()
806 builder_cond0.AddInstruction(HloInstruction::CreateCompare( in TEST_F()
835 builder_cond1.AddInstruction(HloInstruction::CreateCompare( in TEST_F()
Ddynamic_padder.cc164 computation->AddInstruction(HloInstruction::CreateCompare( in Run()
Dpattern_matcher_test.cc938 auto eq = HloInstruction::CreateCompare(shape, p0.get(), p1.get(), in TEST()
940 auto ne = HloInstruction::CreateCompare(shape, p0.get(), p1.get(), in TEST()
944 auto le = HloInstruction::CreateCompare(shape, p0.get(), add.get(), in TEST()
Dhlo_memory_scheduler_test.cc258 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), cond_param, in TEST_F()
Dheap_simulator_test.cc57 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), cond_iter, in TEST_F()
148 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), reshape, in TEST_F()
707 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), cond_iter, in TEST_F()
Dhlo_dce_test.cc227 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), param, in TEST_F()
Dconvolution_group_converter.cc194 return add_instruction(HloInstruction::CreateCompare( in GetExpandedFilterMask()
Dhlo_creation_utils.cc54 HloInstruction::CreateCompare(binary_op_shape, lhs, rhs, direction)); in MakeCompareHlo()
Dtuple_points_to_analysis_test.cc937 auto result = builder.AddInstruction(HloInstruction::CreateCompare( in TEST_F()
1188 builder.AddInstruction(HloInstruction::CreateCompare( in TEST_F()
Dcall_graph_test.cc87 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), param0, in MakeConditionComputation()
Ddynamic_dimension_inference_test.cc70 embedded_builder.AddInstruction(HloInstruction::CreateCompare( in GetGe()
Dbuffer_assignment_test.cc194 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), index, in BuildWhileConditionComputation()
1867 builder.AddInstruction(HloInstruction::CreateCompare( in BuildWhileConditionComputation()
2140 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), param, in TEST_F()
Dhlo_computation_test.cc513 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), param0, in TEST_F()
Dhlo_instruction.h449 static std::unique_ptr<HloInstruction> CreateCompare(
Dhlo_dataflow_analysis_test.cc2270 auto result = builder.AddInstruction(HloInstruction::CreateCompare( in TEST_F()
2594 builder.AddInstruction(HloInstruction::CreateCompare( in TEST_F()
Dhlo_instruction.cc175 CreateCompare(shape, operands(0), operands(1), *comparison_direction); in CreateFromProto()
797 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateCompare( in CreateCompare() function in xla::HloInstruction
Dalgebraic_simplifier.cc876 computation->AddInstruction(HloInstruction::CreateCompare( in TryDivideToShift()
2478 computation->AddInstruction(HloInstruction::CreateCompare( in TryRemainderToAnd()
Dcopy_insertion_test.cc423 builder.AddInstruction(HloInstruction::CreateCompare( in BuildConditionComputation()
Dhlo_evaluator_test.cc2880 b.AddInstruction(HloInstruction::CreateCompare(expected.shape(), c1, c2, in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dgpu_index_test.cc51 builder.AddInstruction(HloInstruction::CreateCompare( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dwhile_transformer_test.cc47 builder.AddInstruction(HloInstruction::CreateCompare( in BuildConditionComputation()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dfusion_test.cc110 root_hlo = HloInstruction::CreateCompare(answer_shape, hlos[1], in TestElementwise2D()

12