| fusion_merger_test.cc | 61 auto one_vec = builder.AddInstruction(HloInstruction::CreateConstant( local
66 data_shape_, HloOpcode::kAdd, one_vec, gte0));
75 // First sub-computation: out1 = Mul(Add(add1, one_vec), one_vec)
77 data_shape_, HloOpcode::kAdd, add1, one_vec));
79 data_shape_, HloOpcode::kMultiply, add2, one_vec));
81 // Second sub-computation: out2 = Add(Mul(add1, one_vec), one_vec)
83 data_shape_, HloOpcode::kMultiply, add1, one_vec));
85 data_shape_, HloOpcode::kAdd, mul0, one_vec));
140 auto one_vec = builder.AddInstruction(HloInstruction::CreateConstant( local
211 auto one_vec = builder.AddInstruction(HloInstruction::CreateConstant( local
[all...] |
