| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | reshape_mover_test.cc | 201 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()
|
| D | comparison_expander.cc | 59 HloInstruction::CreateTernary(signed_shape, HloOpcode::kSelect, in BitcastConvertFloatingPointToIntegral()
|
| D | hlo_alias_analysis_test.cc | 870 auto select11 = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
872 auto select12 = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
874 auto select34 = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
876 auto select1234 = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
963 auto select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
|
| D | tuple_points_to_analysis_test.cc | 421 auto select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
450 auto select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
499 auto select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
535 auto select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
|
| D | stable_sort_expander.cc | 191 comparator->AddInstruction(HloInstruction::CreateTernary( in ExpandInstruction()
|
| D | convolution_group_converter.cc | 347 auto new_filter = add(HloInstruction::CreateTernary( in HandleBatchGroupCount()
476 auto new_filter = add(HloInstruction::CreateTernary( in HandleConvolution()
|
| D | hlo_dataflow_analysis_test.cc | 913 auto select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_P()
950 auto select11 = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_P()
952 auto select12 = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_P()
954 auto select34 = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_P()
956 auto select1234 = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_P()
1028 auto select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_P()
1112 auto select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_P()
|
| D | hlo_cost_analysis_test.cc | 614 HloInstruction::CreateTernary(r2f32, HloOpcode::kClamp, c2, add, add)); in TEST_F()
675 HloInstruction::CreateTernary(r2f32, HloOpcode::kClamp, c2, add, add)); in TEST_F()
|
| D | conditional_simplifier.cc | 179 return computation->AddInstruction(HloInstruction::CreateTernary( in TryRemoveConditional()
|
| D | algebraic_simplifier.cc | 1489 computation->AddInstruction(HloInstruction::CreateTernary( in TryDivideToShift()
1501 return HloInstruction::CreateTernary(divide->shape(), HloOpcode::kSelect, in TryDivideToShift()
2557 HloInstruction::CreateTernary(gather->shape(), HloOpcode::kSelect, in HandleGather()
2599 return HloInstruction::CreateTernary(to_clamp->shape(), HloOpcode::kClamp, in MinMaxToClamp()
2740 multiply, HloInstruction::CreateTernary( in HandleMultiply()
3713 computation->AddInstruction(HloInstruction::CreateTernary( in TryRemainderToAnd()
3725 return HloInstruction::CreateTernary( in TryRemainderToAnd()
5564 auto clamp = HloInstruction::CreateTernary(root->shape(), HloOpcode::kClamp, in TransformToClampIfSameShape()
|
| D | hlo_evaluator_test.cc | 124 b.AddInstruction(HloInstruction::CreateTernary( in TestTernaryOp()
233 HloInstruction::CreateTernary(shape, HloOpcode::kClamp, c1, c2, c3)); in TEST_P()
259 HloInstruction::CreateTernary(shape, HloOpcode::kClamp, c1, c2, c3)); in TEST_P()
281 HloInstruction::CreateTernary(shape, HloOpcode::kClamp, c1, c2, c3)); in TEST_P()
306 HloInstruction::CreateTernary(shape, HloOpcode::kSelect, c1, c2, c3)); in TEST_P()
|
| D | copy_insertion_test.cc | 213 builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
385 HloInstruction* select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
704 auto data_init = builder.AddInstruction(HloInstruction::CreateTernary( in BuildWhileInstruction_InitPointsToAmbiguous()
|
| D | bfloat16_propagation_test.cc | 115 HloInstruction::CreateTernary(shape, HloOpcode::kSelect, pred, c, add1)); in TEST_F()
564 HloInstruction* sel = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
|
| D | hlo_creation_utils.cc | 408 computation->AddInstruction(HloInstruction::CreateTernary( in MakeSelectHlo()
|
| D | dynamic_padder.cc | 271 HloInstruction::CreateTernary(inst->shape(), HloOpcode::kSelect, pred, in PadWithScalar()
1379 update = comp->AddInstruction(HloInstruction::CreateTernary( in RewriteDynamicUpdateSlice()
|
| D | hlo_instruction.h | 563 static std::unique_ptr<HloInstruction> CreateTernary(const Shape& shape,
|
| D | layout_assignment_test.cc | 266 auto select = builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
|
| D | hlo_instruction.cc | 953 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateTernary( in CreateTernary() function in xla::HloInstruction
1769 clone = CreateTernary(shape, opcode_, new_operands[0], new_operands[1], in CloneWithNewOperands()
|
| D | hlo_instruction_test.cc | 871 HloInstruction::CreateTernary(r0f32_, HloOpcode::kClamp, c2, add, add)); in TEST_F()
|
| D | algebraic_simplifier_test.cc | 529 builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
551 builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
571 builder.AddInstruction(HloInstruction::CreateTernary( in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cusolver_rewriter.cc | 124 computation->AddInstruction(HloInstruction::CreateTernary( in CreateCholesky()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | gather_scatter_handler.cc | 241 HloInstruction::CreateTernary(indices.base_shape(), HloOpcode::kClamp, in ParititonTrivialIndexedOperandDimension()
282 auto filtered = b->AddInstruction(HloInstruction::CreateTernary( in ParititonTrivialIndexedOperandDimension()
596 b_.AddInstruction(HloInstruction::HloInstruction::CreateTernary( in HandleScatter()
|
| D | spmd_partitioner.cc | 539 auto result = state_.b->AddInstruction(HloInstruction::CreateTernary( in PadWithValue()
1075 auto operand = state_.b->AddInstruction(HloInstruction::CreateTernary( in Broadcast()
2429 new_indices[dim] = add_hlo(HloInstruction::CreateTernary( in HandleDynamicUpdateSlice()
2444 return add_hlo(HloInstruction::CreateTernary( in HandleDynamicUpdateSlice()
3355 slice_offsets[i] = b_.AddInstruction(HloInstruction::CreateTernary( in HandleSelectAndScatter()
|
| D | spmd_partitioner_util.cc | 1065 HloInstruction::CreateTernary(valid_slice->shape(), HloOpcode::kSelect, in ExchangeHaloAndGetValidData()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | cpu_gpu_fusion_test.cc | 117 root_hlo = HloInstruction::CreateTernary(answer_shape, opcode, hlos[1], in TestElementwise2D()
226 HloInstruction::CreateTernary(ShapeUtil::MakeShape(F32, {2, 3}), in XLA_TEST_F()
|