| /external/tensorflow/tensorflow/compiler/xla/service/ |
| defuser_test.cc | 72 computation->CreateFusionInstruction({add},
98 computation->CreateFusionInstruction({add},
136 computation->CreateFusionInstruction(
175 computation->CreateFusionInstruction({add2, constant, div, mul},
177 computation->CreateFusionInstruction({sub, negate, add},
203 auto outer_fusion = computation->CreateFusionInstruction(
207 outer_fusion->fused_instructions_computation()->CreateFusionInstruction(
|
| hlo_graph_dumper_test.cc | 64 auto* outer_fusion = root_computation->CreateFusionInstruction(
79 outer_fusion->fused_instructions_computation()->CreateFusionInstruction(
|
| hlo_instruction_test.cc | 635 auto* fusion = computation->CreateFusionInstruction(
653 auto* fusion = computation->CreateFusionInstruction(
675 auto* fusion = computation->CreateFusionInstruction(
698 auto* fusion = computation->CreateFusionInstruction(
776 auto* fusion = computation->CreateFusionInstruction(
823 auto* fusion = computation->CreateFusionInstruction(
[all...] |
| tuple_points_to_analysis_test.cc | [all...] |
| hlo_computation.h | 241 HloInstruction* CreateFusionInstruction(
|
| buffer_liveness_test.cc | 654 computation->CreateFusionInstruction(
658 computation->CreateFusionInstruction(
664 computation->CreateFusionInstruction({gte0},
[all...] |
| hlo_dataflow_analysis_test.cc | [all...] |
| hlo_memory_scheduler_test.cc | 225 auto fusion = computation->CreateFusionInstruction(
|
| hlo_cost_analysis_test.cc | 474 auto* fusion = computation->CreateFusionInstruction(
516 auto* fusion = computation->CreateFusionInstruction(
|
| hlo_cse_test.cc | 509 ->CreateFusionInstruction({mul, add1, add2},
|
| reshape_mover_test.cc | 403 computation->CreateFusionInstruction({add},
|
| hlo_computation.cc | 614 HloInstruction* HloComputation::CreateFusionInstruction(
[all...] |
| layout_assignment_test.cc | 185 auto fusion = computation->CreateFusionInstruction(
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| fusion_test.cc | 126 ->CreateFusionInstruction(
231 // CreateFusionInstruction needs the `instructions_to_fuse` argument in
235 ->CreateFusionInstruction(
260 // CreateFusionInstruction needs `instructions_to_fuse` in reverse topological
263 ->CreateFusionInstruction(/*instructions_to_fuse=*/{add3, const2},
295 ->CreateFusionInstruction(/*instructions_to_fuse=*/{y, x, two},
323 ->CreateFusionInstruction(/*instructions_to_fuse=*/{add2, broadcast},
339 ->CreateFusionInstruction(/*instructions_to_fuse=*/{reshape},
354 ->CreateFusionInstruction(/*instructions_to_fuse=*/{reshape1},
369 ->CreateFusionInstruction(/*instructions_to_fuse=*/{reshape1}
[all...] |
| multioutput_fusion_test.cc | 107 computation->CreateFusionInstruction(
170 CHECK_NE(computation->CreateFusionInstruction(
|
