| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_fusible_test.cc | 49 ASSERT_EQ(reduce->opcode(), HloOpcode::kReduce); in TEST_F()
86 HloOpcode::kReduce); in TEST_F()
110 ASSERT_EQ(reduce->fused_expression_root()->opcode(), HloOpcode::kReduce); in TEST_F()
146 HloOpcode::kReduce); in TEST_F()
174 ASSERT_EQ(reduce->opcode(), HloOpcode::kReduce); in TEST_F()
193 ASSERT_EQ(reduce->opcode(), HloOpcode::kReduce); in TEST_F()
210 ASSERT_EQ(reduce->opcode(), HloOpcode::kReduce); in TEST_F()
|
| D | ir_emitter_unnested.cc | 629 case HloOpcode::kReduce: { in HandleFusion()
633 if (root->opcode() == HloOpcode::kReduce && root->shape().IsTuple()) { in HandleFusion()
1915 case HloOpcode::kReduce: in BuildInitializerThunk()
1920 CHECK(inst->operand(index.back())->opcode() == HloOpcode::kReduce) in BuildInitializerThunk()
2229 case HloOpcode::kReduce: in MayPreventVectorization()
2624 CHECK(opcode == HloOpcode::kReduce || opcode == HloOpcode::kTuple); in GetOutputInstructions()
2634 return inst->opcode() == HloOpcode::kReduce; in GetFirstReduceInstruction()
2713 if (output_instructions[i]->opcode() != HloOpcode::kReduce) { in EmitPrologueForReduction()
2901 if (inst->opcode() == HloOpcode::kReduce) { in EmitTileElementForReduction()
3468 if (inst->opcode() == HloOpcode::kReduce) { in AreFusedReductionOutputsConsistent()
[all …]
|
| D | gpu_fusible.cc | 174 (instr.opcode() == HloOpcode::kReduce && in IsLoopFusible()
|
| D | ir_emission_utils.cc | 161 if (HloOpcode::kReduce != reduce.opcode()) { in IsReductionToVector()
|
| D | elemental_ir_emitter.cc | 440 case HloOpcode::kReduce: in MakeElementGenerator()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_opcode_test.cc | 63 case HloOpcode::kReduce: in TEST()
|
| D | bfloat16_support.cc | 111 case HloOpcode::kReduce: in EffectiveOperandPrecisionIsOutputPrecision()
|
| D | dynamic_padder.cc | 51 case HloOpcode::kReduce: in ChooseIdentityValue()
|
| D | hlo_opcode.h | 115 V(kReduce, "reduce", kHloOpcodeIsVariadic) \
|
| D | bfloat16_normalization_test.cc | 41 hlo.opcode() == HloOpcode::kReduce || in SupportsBF16Operand()
54 if (hlo.opcode() == HloOpcode::kAdd || hlo.opcode() == HloOpcode::kReduce || in SupportsBF16Output()
|
| D | hlo_element_type_converter.cc | 146 opcode == HloOpcode::kMap || opcode == HloOpcode::kReduce || in Run()
|
| D | hlo_instruction.cc | 234 case HloOpcode::kReduce: in CreateFromProto()
1360 case HloOpcode::kReduce: in CloneWithNewOperands()
1810 case HloOpcode::kReduce: in IdenticalSlowPath()
1993 case HloOpcode::kReduce: in to_apply()
2013 case HloOpcode::kReduce: in set_to_apply()
2308 opcode() == HloOpcode::kReduce || in ExtraAttributesToString()
2355 case HloOpcode::kReduce: in ExtraAttributesToString()
2563 case HloOpcode::kReduce: in Visit()
2939 case HloOpcode::kReduce: in OperandElementUse()
|
| D | call_graph.cc | 63 case HloOpcode::kReduce: in GetInstructionCallContext()
|
| D | instruction_fusion.cc | 146 case HloOpcode::kReduce: in IsExpensive()
|
| D | hlo_module.cc | 137 case HloOpcode::kReduce: in ReplaceComputations()
|
| D | layout_assignment.cc | 1369 /*mandatory=*/user->opcode() == HloOpcode::kReduce, in PropagateOperandConstraint()
2066 case HloOpcode::kReduce: in InstructionCanChangeLayout()
|
| D | hlo_graph_dumper.cc | 1031 case HloOpcode::kReduce: in GetInstructionColor()
|
| D | hlo_verifier.cc | 49 case HloOpcode::kReduce: in IsCallerInstruction()
|
| D | buffer_assignment.cc | 190 case HloOpcode::kReduce: in GatherComputationsByAllocationType()
|
| D | hlo_instructions.cc | 731 : HloInstruction(HloOpcode::kReduce, shape), in HloReduceInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reduce_hlo_test.cc | 82 ASSERT_EQ(reduce_instruction->opcode(), HloOpcode::kReduce); in XLA_TEST_P()
|
| D | test_utils.cc | 376 (opcode == HloOpcode::kReduce && in NeedsInitValue()
483 case HloOpcode::kReduce: in CreateLiteralForConstrainedUses()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager.cc | 371 AddParticipant(std::move(participant), context, kReduce, reduction_op); in AddReduceSend()
377 AddParticipant(std::move(participant), context, kReduce, reduction_op); in AddReduceRecv()
591 case kReduce: { in LoopKernelLaunches()
|
| D | nccl_manager.h | 176 kReduce = 3, enumerator
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | elemental_ir_emitter.cc | 136 case HloOpcode::kReduce: in MakeElementGenerator()
|