diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-08-10 16:32:55 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-08-10 16:36:36 -0700 |
| commit | 822603aed3f20159f06284af5ce35efa81b95ed6 (patch) | |
| tree | 9343f500cc6dcbab6617959aa9542985e28a7c87 /tensorflow/compiler/xla/service/hlo_computation.h | |
| parent | c035aa2a8a6e3a35c68495bc4ff68a7bbd60d57d (diff) | |
Merging sibling fusion instruction using multi_output_fusion
PiperOrigin-RevId: 164920220
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_computation.h')
| -rw-r--r-- | tensorflow/compiler/xla/service/hlo_computation.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_computation.h b/tensorflow/compiler/xla/service/hlo_computation.h index 5f651060fd..0a33d0c1cf 100644 --- a/tensorflow/compiler/xla/service/hlo_computation.h +++ b/tensorflow/compiler/xla/service/hlo_computation.h @@ -112,8 +112,8 @@ class HloComputation { // Set the root of the computation to the given instruction. The instruction // must have already been added to the computation and have the same shape as - // the result of the computation. - void set_root_instruction(HloInstruction* instruction); + // the result of the computation for non fusion computations. + void set_root_instruction(HloInstruction* new_root_instruction); // Return the root instruction of the computation. The root instruction is the // instruction which produces the output of the computation. |