Searched refs:CreateFusionInstruction (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | fusion_test.cc | 126 ->CreateFusionInstruction( in TestElementwise2D()
235 ->CreateFusionInstruction( in XLA_TEST_F()
263 ->CreateFusionInstruction(/*instructions_to_fuse=*/{add3, const2}, in XLA_TEST_F()
295 ->CreateFusionInstruction(/*instructions_to_fuse=*/{y, x, two}, in XLA_TEST_F()
323 ->CreateFusionInstruction(/*instructions_to_fuse=*/{add2, broadcast}, in XLA_TEST_F()
339 ->CreateFusionInstruction(/*instructions_to_fuse=*/{reshape}, in XLA_TEST_F()
354 ->CreateFusionInstruction(/*instructions_to_fuse=*/{reshape1}, in XLA_TEST_F()
369 ->CreateFusionInstruction(/*instructions_to_fuse=*/{reshape1}, in XLA_TEST_F()
384 ->CreateFusionInstruction(/*instructions_to_fuse=*/{reshape1}, in XLA_TEST_F()
399 ->CreateFusionInstruction(/*instructions_to_fuse=*/{reshape1}, in XLA_TEST_F()
[all …]
|
| D | multioutput_fusion_test.cc | 107 computation->CreateFusionInstruction( in RunTest2D()
170 CHECK_NE(computation->CreateFusionInstruction( in RunTest1D()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | defuser_test.cc | 72 computation->CreateFusionInstruction({add}, in TEST_F()
98 computation->CreateFusionInstruction({add}, in TEST_F()
136 computation->CreateFusionInstruction( in TEST_F()
175 computation->CreateFusionInstruction({add2, constant, div, mul}, in TEST_F()
177 computation->CreateFusionInstruction({sub, negate, add}, in TEST_F()
203 auto outer_fusion = computation->CreateFusionInstruction( in TEST_F()
207 outer_fusion->fused_instructions_computation()->CreateFusionInstruction( in TEST_F()
|
| D | hlo_instruction_test.cc | 635 auto* fusion = computation->CreateFusionInstruction( in TEST_F()
653 auto* fusion = computation->CreateFusionInstruction( in TEST_F()
675 auto* fusion = computation->CreateFusionInstruction( in TEST_F()
698 auto* fusion = computation->CreateFusionInstruction( in TEST_F()
776 auto* fusion = computation->CreateFusionInstruction( in TEST_F()
823 auto* fusion = computation->CreateFusionInstruction( in TEST_F()
1073 HloInstruction* fusion = computation->CreateFusionInstruction( in TEST_F()
1114 HloInstruction* fusion = computation->CreateFusionInstruction( in TEST_F()
1157 HloInstruction* fusion = computation->CreateFusionInstruction( in TEST_F()
1198 HloInstruction* fusion = computation->CreateFusionInstruction( in TEST_F()
[all …]
|
| D | hlo_graph_dumper_test.cc | 64 auto* outer_fusion = root_computation->CreateFusionInstruction( in TEST_F()
79 outer_fusion->fused_instructions_computation()->CreateFusionInstruction( in TEST_F()
|
| D | tuple_points_to_analysis_test.cc | 896 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
990 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
1144 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
1172 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
1239 sub_computation->CreateFusionInstruction({add, ones}, in TEST_F()
1274 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
|
| D | hlo_dataflow_analysis_test.cc | 2013 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2056 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2108 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2137 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2168 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2193 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2324 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2367 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2523 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
2551 auto fusion = computation_->CreateFusionInstruction( in TEST_F()
[all …]
|
| D | buffer_liveness_test.cc | 654 computation->CreateFusionInstruction( in BuildModule()
658 computation->CreateFusionInstruction( in BuildModule()
664 computation->CreateFusionInstruction({gte0}, in BuildModule()
|
| D | hlo_cost_analysis_test.cc | 474 auto* fusion = computation->CreateFusionInstruction( in TEST_F()
516 auto* fusion = computation->CreateFusionInstruction( in TEST_F()
|
| D | hlo_computation.h | 241 HloInstruction* CreateFusionInstruction(
|
| D | hlo_memory_scheduler_test.cc | 225 auto fusion = computation->CreateFusionInstruction( in TEST_F()
|
| D | reshape_mover_test.cc | 403 computation->CreateFusionInstruction({add}, in TEST_F()
|
| D | hlo_cse_test.cc | 509 ->CreateFusionInstruction({mul, add1, add2}, in TEST_F()
|
| D | hlo_computation.cc | 614 HloInstruction* HloComputation::CreateFusionInstruction( in CreateFusionInstruction() function in xla::HloComputation
|
| D | layout_assignment_test.cc | 185 auto fusion = computation->CreateFusionInstruction( in TEST_F()
|