Searched defs:instruction_to_fuse (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_instruction.h959 // Fuses the given instruction in this fusion instruction. instruction_to_fuse
961 // instruction. instruction_to_fuse is unchanged. Instruction is cloned rather
968 HloInstruction* FuseInstruction(HloInstruction* instruction_to_fuse) { argument
969 return FuseInstructionInternal(instruction_to_fuse);
973 // multioutput fusion instruction. A clone of the instruction_to_fuse will
975 // instruction_to_fuse will be redirected to this fusion instructions.
976 // instruction_to_fuse will be removed from its parent computation.
980 HloInstruction* instruction_to_fuse) {
981 return FuseInstructionInternal(instruction_to_fuse, /* add_output */ true);
1308 // is false (which is the default), instruction_to_fuse i
979 FuseInstructionIntoMultiOutput( HloInstruction* instruction_to_fuse) argument
[all...]
H A Dhlo_instruction.cc920 HloInstruction* instruction_to_fuse, bool add_output) {
924 // instruction_to_fuse.
926 CHECK(IsUserOf(instruction_to_fuse));
929 CloneAndFuseInternal(instruction_to_fuse, add_output);
934 HloInstruction* instruction_to_fuse, bool add_output) {
936 CHECK(instruction_to_fuse->IsFusable()) << instruction_to_fuse->ToString();
937 VLOG(3) << "CloneAndFuseInternal:\n" << instruction_to_fuse->ToString();
943 builder.AddInstruction(instruction_to_fuse->Clone(/*suffix=*/""));
949 instruction_to_fuse
919 FuseInstructionInternal( HloInstruction* instruction_to_fuse, bool add_output) argument
933 CloneAndFuseInternal( HloInstruction* instruction_to_fuse, bool add_output) argument
[all...]

Completed in 183 milliseconds