| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | cpu_gpu_fusion_test.cc | 125 HloInstruction::FusionKind::kLoop); in TestElementwise2D()
235 HloInstruction::FusionKind::kLoop); in XLA_TEST_F()
261 HloInstruction::FusionKind::kLoop); in XLA_TEST_F()
293 HloInstruction::FusionKind::kLoop); in XLA_TEST_F()
321 HloInstruction::FusionKind::kLoop); in XLA_TEST_F()
337 HloInstruction::FusionKind::kLoop); in XLA_TEST_F()
352 HloInstruction::FusionKind::kLoop); in XLA_TEST_F()
367 HloInstruction::FusionKind::kLoop); in XLA_TEST_F()
382 HloInstruction::FusionKind::kLoop); in XLA_TEST_F()
397 HloInstruction::FusionKind::kLoop); in XLA_TEST_F()
[all …]
|
| D | multioutput_fusion_test.cc | 106 {tuple, sub, add2, broadcast}, HloInstruction::FusionKind::kLoop), in RunTest2D()
170 HloInstruction::FusionKind::kLoop), in RunTest1D()
|
| /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 | 639 {exp}, HloInstruction::FusionKind::kLoop); in TEST_F()
657 {add}, HloInstruction::FusionKind::kLoop); in TEST_F()
679 {exp3, exp2, exp1}, HloInstruction::FusionKind::kLoop); in TEST_F()
702 {exp2, exp1}, HloInstruction::FusionKind::kLoop); in TEST_F()
936 {map_3_y}, HloInstruction::FusionKind::kLoop); in TEST_F()
983 {tuple, sub, mul, exp, clamp, add}, HloInstruction::FusionKind::kLoop); in TEST_F()
1234 {max, broadcast, div, mul}, HloInstruction::FusionKind::kLoop); in TEST_F()
1275 {sub, broadcast, min}, HloInstruction::FusionKind::kLoop); in TEST_F()
1318 {dot, reshape}, HloInstruction::FusionKind::kLoop); in TEST_F()
1417 {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 | 502 enum class FusionKind { enum
1091 const Shape& shape, FusionKind fusion_kind, HloInstruction* fused_root);
1094 const Shape& shape, FusionKind fusion_kind,
1976 FusionKind fusion_kind() const;
1979 void set_fusion_kind(FusionKind kind);
2404 std::string ToString(HloInstruction::FusionKind kind);
2405 StatusOr<HloInstruction::FusionKind> StringToFusionKind(
2430 std::ostream& operator<<(std::ostream& os, HloInstruction::FusionKind kind);
|
| D | pattern_matcher_test.cc | 152 TEST_F(PatternMatcherTest, FusionKind) { in TEST_F() argument
169 root, match::Op().WithFusionKind(HloInstruction::FusionKind::kLoop))); in TEST_F()
171 root, match::Op().WithFusionKind(HloInstruction::FusionKind::kInput))); in TEST_F()
173 HloInstruction::FusionKind::kLoop))); in TEST_F()
808 iota, m::Op().WithFusionKind(HloInstruction::FusionKind::kLoop), in TEST_F()
|
| D | hlo_instruction.cc | 410 TF_ASSIGN_OR_RETURN(FusionKind fusion_kind, in CreateFromProto()
1784 const Shape& shape, FusionKind fusion_kind, HloInstruction* fused_root) { in CreateFusion()
1789 const Shape& shape, FusionKind fusion_kind, in CreateFusion()
3424 return opcode() == HloOpcode::kFusion && fusion_kind() == FusionKind::kInput; in IsInputFusion()
3428 return opcode() == HloOpcode::kFusion && fusion_kind() == FusionKind::kLoop; in IsLoopFusion()
3432 return opcode() == HloOpcode::kFusion && fusion_kind() == FusionKind::kOutput; in IsOutputFusion()
3436 return opcode() == HloOpcode::kFusion && fusion_kind() == FusionKind::kCustom; in IsCustomFusion()
4025 std::string ToString(HloInstruction::FusionKind kind) { in ToString()
4027 case HloInstruction::FusionKind::kLoop: in ToString()
4029 case HloInstruction::FusionKind::kInput: in ToString()
[all …]
|
| D | bfloat16_propagation_test.cc | 400 tuple_f0->shape(), HloInstruction::FusionKind::kCustom, {add, add}, in TEST_F()
413 dot->shape(), HloInstruction::FusionKind::kCustom, {fusion0}, comp_f1)); in TEST_F()
445 f32_shape, HloInstruction::FusionKind::kLoop, {param}, comp_f)); in TEST_F()
479 dot_f->shape(), HloInstruction::FusionKind::kCustom, {add, add}, comp_f)); in TEST_F()
518 tuple_f->shape(), HloInstruction::FusionKind::kCustom, {add, add}, in TEST_F()
649 shape, HloInstruction::FusionKind::kCustom, {body_transpose}, comp_f)); in TEST_F()
|
| D | instruction_fusion.h | 212 virtual HloInstruction::FusionKind ChooseKind(const HloInstruction* producer,
|
| D | hlo_dataflow_analysis_test.cc | 2127 HloInstruction::FusionKind::kLoop); in TEST_F()
2171 HloInstruction::FusionKind::kLoop); in TEST_F()
2225 {reverse, neg}, HloInstruction::FusionKind::kLoop); in TEST_F()
2255 {tuple, copy1, copy0}, HloInstruction::FusionKind::kLoop); in TEST_F()
2287 {exp, neg}, HloInstruction::FusionKind::kLoop); in TEST_F()
2313 {dus, ds, zero}, HloInstruction::FusionKind::kLoop); in TEST_F()
2420 HloInstruction::FusionKind::kLoop); in TEST_F()
2464 HloInstruction::FusionKind::kLoop); in TEST_F()
2703 {add, dot}, HloInstruction::FusionKind::kOutput); in TEST_F()
2732 {add, two, reverse}, HloInstruction::FusionKind::kOutput); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion.cc | 193 HloInstruction::FusionKind CpuInstructionFusion::ChooseKind( in ChooseKind()
196 ? HloInstruction::FusionKind::kOutput in ChooseKind()
197 : HloInstruction::FusionKind::kLoop; in ChooseKind()
|
| D | cpu_instruction_fusion.h | 44 HloInstruction::FusionKind ChooseKind(
|
| D | cpu_instruction_fusion_test.cc | 248 HloInstruction::FusionKind fusion_kind = in RunFusionAndCheckOpcodesWereFused()
249 HloInstruction::FusionKind::kLoop) { in RunFusionAndCheckOpcodesWereFused()
662 HloInstruction::FusionKind::kOutput); in TEST_F()
675 HloInstruction::FusionKind::kOutput); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_fusible.cc | 158 instr.fusion_kind() == HloInstruction::FusionKind::kInput && in IsInputFusibleScatter()
186 instr.fusion_kind() == HloInstruction::FusionKind::kLoop) || in IsLoopFusible()
569 HloInstruction::FusionKind ChooseFusionKind(const HloInstruction& /*producer*/, in ChooseFusionKind()
571 return IsInputFusible(consumer) ? HloInstruction::FusionKind::kInput in ChooseFusionKind()
572 : HloInstruction::FusionKind::kLoop; in ChooseFusionKind()
|
| D | hlo_fusion_stats.cc | 123 if (fusion->fusion_kind() == HloInstruction::FusionKind::kLoop) { in HandleFusion()
126 } else if (fusion->fusion_kind() == HloInstruction::FusionKind::kInput) { in HandleFusion()
|
| D | instruction_fusion.h | 46 HloInstruction::FusionKind ChooseKind(
|
| D | gpu_fusible.h | 122 HloInstruction::FusionKind ChooseFusionKind(const HloInstruction& producer,
|
| D | instruction_fusion.cc | 122 HloInstruction::FusionKind GpuInstructionFusion::ChooseKind( in ChooseKind()
|
| D | instruction_fusion_test.cc | 388 EXPECT_EQ(root->fusion_kind(), HloInstruction::FusionKind::kLoop); in TEST_F()
530 HloInstruction::FusionKind::kInput); in TEST_F()
695 EXPECT_EQ(root->fusion_kind(), HloInstruction::FusionKind::kInput); in TEST_F()
|
| D | horizontal_input_fusion.cc | 130 HloInstruction::FusionKind::kInput)); in Run()
|
| D | horizontal_loop_fusion.cc | 456 HloInstruction::FusionKind::kInput, in Fuse()
522 MakeFusionInstruction(instr, HloInstruction::FusionKind::kLoop)); in Run()
|
| D | fusion_merger_test.cc | 352 shape, HloInstruction::FusionKind::kLoop, params, subcomp); in TEST_F()
|
| D | fusion_bitcast_lift.cc | 84 if (fusion->fusion_kind() != HloInstruction::FusionKind::kInput || in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | parallel_reduction_test.cc | 169 HloInstruction::FusionKind::kInput, entry_params, fusion_computation)); in TEST_F()
|