Home
last modified time | relevance | path

Searched refs:CreateTernary (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dreshape_mover_test.cc201 builder.AddInstruction(HloInstruction::CreateTernary(
282 builder.AddInstruction(HloInstruction::CreateTernary(
434 builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
464 auto select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
Dhlo_alias_analysis_test.cc878 auto select11 = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
880 auto select12 = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
882 auto select34 = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
884 auto select1234 = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
971 auto select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
Ddynamic_padder.cc172 computation->AddInstruction(HloInstruction::CreateTernary( in Run()
Dconvolution_group_converter.cc250 auto new_filter = add(HloInstruction::CreateTernary( in HandleBatchGroupCount()
376 auto new_filter = add(HloInstruction::CreateTernary( in HandleConvolution()
Dstable_sort_expander.cc191 comparator->AddInstruction(HloInstruction::CreateTernary( in ExpandInstruction()
Dtuple_points_to_analysis_test.cc393 auto select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
422 auto select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
471 auto select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
507 auto select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
Dhlo_dataflow_analysis_test.cc860 auto select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_P()
897 auto select11 = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_P()
899 auto select12 = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_P()
901 auto select34 = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_P()
903 auto select1234 = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_P()
975 auto select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_P()
1059 auto select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_P()
Dhlo_evaluator_test.cc123 b.AddInstruction(HloInstruction::CreateTernary( in TestTernaryOp()
163 HloInstruction::CreateTernary(shape, HloOpcode::kClamp, c1, c2, c3)); in TEST_P()
189 HloInstruction::CreateTernary(shape, HloOpcode::kClamp, c1, c2, c3)); in TEST_P()
211 HloInstruction::CreateTernary(shape, HloOpcode::kClamp, c1, c2, c3)); in TEST_P()
236 HloInstruction::CreateTernary(shape, HloOpcode::kSelect, c1, c2, c3)); in TEST_P()
Dhlo_creation_utils.cc284 return computation->AddInstruction(HloInstruction::CreateTernary( in MakeSelectHlo()
Dcopy_insertion_test.cc213 builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
385 HloInstruction* select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
698 auto data_init = builder.AddInstruction(HloInstruction::CreateTernary( in BuildWhileInstruction_InitPointsToAmbiguous()
Dalgebraic_simplifier.cc884 computation->AddInstruction(HloInstruction::CreateTernary( in TryDivideToShift()
906 return HloInstruction::CreateTernary(divide->shape(), HloOpcode::kSelect, in TryDivideToShift()
2486 computation->AddInstruction(HloInstruction::CreateTernary( in TryRemainderToAnd()
2505 return HloInstruction::CreateTernary( in TryRemainderToAnd()
3685 auto clamp = HloInstruction::CreateTernary(root->shape(), HloOpcode::kClamp, in TransformToClampIfSameShape()
Dbfloat16_propagation_test.cc115 HloInstruction::CreateTernary(shape, HloOpcode::kSelect, pred, c, add1)); in TEST_F()
534 HloInstruction* sel = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
Dhlo_cost_analysis_test.cc463 HloInstruction::CreateTernary(r2f32, HloOpcode::kClamp, c2, add, add)); in TEST_F()
Dhlo_instruction.h416 static std::unique_ptr<HloInstruction> CreateTernary(const Shape& shape,
Dlayout_assignment_test.cc266 auto select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
Dhlo_instruction.cc750 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateTernary( in CreateTernary() function in xla::HloInstruction
1445 clone = CreateTernary(shape, opcode_, new_operands[0], new_operands[1], in CloneWithNewOperands()
Dhlo_instruction_test.cc811 HloInstruction::CreateTernary(r0f32_, HloOpcode::kClamp, c2, add, add)); in TEST_F()
Dalgebraic_simplifier_test.cc308 builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
330 builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
350 builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
Dbuffer_assignment_test.cc1523 HloInstruction::CreateTernary(tuple_shape, HloOpcode::kTupleSelect, in TEST_F()
Dhlo_parser.cc794 instruction = builder->AddInstruction(HloInstruction::CreateTernary( in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dfusion_test.cc118 root_hlo = HloInstruction::CreateTernary(answer_shape, opcode, hlos[1], in TestElementwise2D()
227 HloInstruction::CreateTernary(ShapeUtil::MakeShape(F32, {2, 3}), in XLA_TEST_F()