| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | cpu_gpu_fusion_test.cc | 127 HloInstruction::FusionKind::kLoop); in TestElementwise2D()
237 HloInstruction::FusionKind::kLoop); in XLA_TEST_F()
263 HloInstruction::FusionKind::kLoop); in XLA_TEST_F()
295 HloInstruction::FusionKind::kLoop); in XLA_TEST_F()
323 HloInstruction::FusionKind::kLoop); in XLA_TEST_F()
339 HloInstruction::FusionKind::kLoop); in XLA_TEST_F()
354 HloInstruction::FusionKind::kLoop); in XLA_TEST_F()
369 HloInstruction::FusionKind::kLoop); in XLA_TEST_F()
384 HloInstruction::FusionKind::kLoop); in XLA_TEST_F()
399 HloInstruction::FusionKind::kLoop); in XLA_TEST_F()
[all …]
|
| D | multioutput_fusion_test.cc | 108 {tuple, sub, add2, broadcast}, HloInstruction::FusionKind::kLoop), in RunTest2D()
172 HloInstruction::FusionKind::kLoop), in RunTest1D()
|
| /external/llvm-project/llvm/lib/Target/PowerPC/ |
| D | PPCMacroFusion.cpp | 26 enum FusionKind { enum in __anonb447b9d50111::FusionFeature
36 FusionKind Kd;
49 FusionFeature(FusionKind Kind, bool HasFeature, int Index, in FusionFeature()
63 FusionKind getKind() const { return Kd; } in getKind()
80 static bool checkOpConstraints(FusionFeature::FusionKind Kd, in checkOpConstraints()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | defuser_test.cc | 73 HloInstruction::FusionKind::kLoop); in TEST_F()
99 HloInstruction::FusionKind::kLoop); in TEST_F()
138 HloInstruction::FusionKind::kLoop); in TEST_F()
176 HloInstruction::FusionKind::kLoop); in TEST_F()
178 HloInstruction::FusionKind::kLoop); in TEST_F()
204 {negate, add}, HloInstruction::FusionKind::kLoop); in TEST_F()
208 {fused_negate}, HloInstruction::FusionKind::kLoop); in TEST_F()
|
| D | hlo_instruction_test.cc | 638 {exp}, HloInstruction::FusionKind::kLoop); in TEST_F()
656 {add}, HloInstruction::FusionKind::kLoop); in TEST_F()
678 {exp3, exp2, exp1}, HloInstruction::FusionKind::kLoop); in TEST_F()
701 {exp2, exp1}, HloInstruction::FusionKind::kLoop); in TEST_F()
837 {map_3_y}, HloInstruction::FusionKind::kLoop); in TEST_F()
884 {tuple, sub, mul, exp, clamp, add}, HloInstruction::FusionKind::kLoop); in TEST_F()
1135 {max, broadcast, div, mul}, HloInstruction::FusionKind::kLoop); in TEST_F()
1176 {sub, broadcast, min}, HloInstruction::FusionKind::kLoop); in TEST_F()
1219 {dot, reshape}, HloInstruction::FusionKind::kLoop); in TEST_F()
1318 {dot, reshape}, HloInstruction::FusionKind::kLoop); in TEST_F()
[all …]
|
| D | hlo_graph_dumper_test.cc | 65 {sums[3], sums[2], sums[1], sums[0]}, HloInstruction::FusionKind::kLoop); in TEST_F()
80 {fused_sums[1], fused_sums[0]}, HloInstruction::FusionKind::kLoop); in TEST_F()
|
| D | hlo_instruction.h | 476 enum class FusionKind { enum
968 const Shape& shape, FusionKind fusion_kind, HloInstruction* fused_root);
971 const Shape& shape, FusionKind fusion_kind,
1760 FusionKind fusion_kind() const;
1763 void set_fusion_kind(FusionKind kind);
2156 string ToString(HloInstruction::FusionKind kind);
2157 StatusOr<HloInstruction::FusionKind> StringToFusionKind(
2176 std::ostream& operator<<(std::ostream& os, HloInstruction::FusionKind kind);
|
| D | instruction_fusion.h | 91 virtual HloInstruction::FusionKind ChooseKind(const HloInstruction* producer,
|
| D | hlo_instruction.cc | 353 TF_ASSIGN_OR_RETURN(FusionKind fusion_kind, in CreateFromProto()
1485 const Shape& shape, FusionKind fusion_kind, HloInstruction* fused_root) { in CreateFusion()
1491 const Shape& shape, FusionKind fusion_kind, in CreateFusion()
3003 return opcode() == HloOpcode::kFusion && fusion_kind() == FusionKind::kInput; in IsInputFusion()
3007 return opcode() == HloOpcode::kFusion && fusion_kind() == FusionKind::kLoop; in IsLoopFusion()
3011 return opcode() == HloOpcode::kFusion && fusion_kind() == FusionKind::kOutput; in IsOutputFusion()
3015 return opcode() == HloOpcode::kFusion && fusion_kind() == FusionKind::kCustom; in IsCustomFusion()
3633 string ToString(HloInstruction::FusionKind kind) { in ToString()
3635 case HloInstruction::FusionKind::kLoop: in ToString()
3637 case HloInstruction::FusionKind::kInput: in ToString()
[all …]
|
| D | pattern_matcher_test.cc | 153 TEST_F(PatternMatcherTest, FusionKind) { in TEST_F() argument
170 root, match::Op().WithFusionKind(HloInstruction::FusionKind::kLoop))); in TEST_F()
172 root, match::Op().WithFusionKind(HloInstruction::FusionKind::kInput))); in TEST_F()
174 HloInstruction::FusionKind::kLoop))); in TEST_F()
732 iota, m::Op().WithFusionKind(HloInstruction::FusionKind::kLoop), in TEST_F()
|
| D | bfloat16_propagation_test.cc | 399 tuple_f0->shape(), HloInstruction::FusionKind::kCustom, {add, add}, in TEST_F()
412 dot->shape(), HloInstruction::FusionKind::kCustom, {fusion0}, comp_f1)); in TEST_F()
444 f32_shape, HloInstruction::FusionKind::kLoop, {param}, comp_f)); in TEST_F()
478 dot_f->shape(), HloInstruction::FusionKind::kCustom, {add, add}, comp_f)); in TEST_F()
517 tuple_f->shape(), HloInstruction::FusionKind::kCustom, {add, add}, in TEST_F()
692 shape, HloInstruction::FusionKind::kCustom, {body_transpose}, comp_f)); in TEST_F()
|
| D | hlo_dataflow_analysis_test.cc | 2141 HloInstruction::FusionKind::kLoop); in TEST_F()
2184 HloInstruction::FusionKind::kLoop); in TEST_F()
2235 {reverse, neg}, HloInstruction::FusionKind::kLoop); in TEST_F()
2264 {tuple, copy1, copy0}, HloInstruction::FusionKind::kLoop); in TEST_F()
2295 {exp, neg}, HloInstruction::FusionKind::kLoop); in TEST_F()
2320 {dus, ds, zero}, HloInstruction::FusionKind::kLoop); in TEST_F()
2422 HloInstruction::FusionKind::kLoop); in TEST_F()
2465 HloInstruction::FusionKind::kLoop); in TEST_F()
2641 {add, dot}, HloInstruction::FusionKind::kOutput); in TEST_F()
2669 {add, two, reverse}, HloInstruction::FusionKind::kOutput); in TEST_F()
[all …]
|
| D | instruction_fusion.cc | 719 HloInstruction::FusionKind InstructionFusion::ChooseKind( in ChooseKind()
721 return HloInstruction::FusionKind::kLoop; in ChooseKind()
|
| D | hlo_cost_analysis_test.cc | 626 {sub, mul, exp, clamp, add}, HloInstruction::FusionKind::kLoop); in TEST_F()
688 {tuple2, sub, mul, exp, clamp, add}, HloInstruction::FusionKind::kLoop); in TEST_F()
742 {add, broadcast}, HloInstruction::FusionKind::kLoop); in TEST_F()
|
| D | hlo_instructions.h | 891 explicit HloFusionInstruction(const Shape& shape, FusionKind fusion_kind,
894 explicit HloFusionInstruction(const Shape& shape, FusionKind fusion_kind,
973 FusionKind fusion_kind() const { return fusion_kind_; } in fusion_kind()
975 void set_fusion_kind(FusionKind kind) { fusion_kind_ = kind; } in set_fusion_kind()
1012 FusionKind fusion_kind_;
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_fusible.cc | 197 instr.fusion_kind() == HloInstruction::FusionKind::kInput && in IsInputFusibleScatter()
225 instr.fusion_kind() == HloInstruction::FusionKind::kLoop) || in IsLoopFusible()
519 HloInstruction::FusionKind ChooseFusionKind(const HloInstruction& /*producer*/, in ChooseFusionKind()
521 return IsInputFusible(consumer) ? HloInstruction::FusionKind::kInput in ChooseFusionKind()
522 : HloInstruction::FusionKind::kLoop; in ChooseFusionKind()
|
| D | instruction_fusion.h | 39 HloInstruction::FusionKind ChooseKind(
|
| D | gpu_fusible.h | 105 HloInstruction::FusionKind ChooseFusionKind(const HloInstruction& producer,
|
| D | instruction_fusion.cc | 128 HloInstruction::FusionKind GpuInstructionFusion::ChooseKind( in ChooseKind()
|
| D | fusion_merger_test.cc | 352 shape, HloInstruction::FusionKind::kLoop, params, subcomp); in TEST_F()
|
| D | instruction_fusion_test.cc | 388 EXPECT_EQ(root->fusion_kind(), HloInstruction::FusionKind::kLoop); in TEST_F()
530 HloInstruction::FusionKind::kInput); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion.cc | 198 HloInstruction::FusionKind CpuInstructionFusion::ChooseKind( in ChooseKind()
201 ? HloInstruction::FusionKind::kOutput in ChooseKind()
202 : HloInstruction::FusionKind::kLoop; in ChooseKind()
|
| D | cpu_instruction_fusion.h | 40 HloInstruction::FusionKind ChooseKind(
|
| D | cpu_instruction_fusion_test.cc | 266 HloInstruction::FusionKind fusion_kind = in RunFusionAndCheckOpcodesWereFused()
267 HloInstruction::FusionKind::kLoop) { in RunFusionAndCheckOpcodesWereFused()
680 HloInstruction::FusionKind::kOutput); in TEST_F()
693 HloInstruction::FusionKind::kOutput); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | parallel_reduction_test.cc | 126 HloInstruction::FusionKind::kInput, entry_params, fusion_computation)); in TEST_F()
|