| 224 | } |
| 225 | |
| 226 | xla::XlaComputation AddAndTuple() { |
| 227 | xla::XlaBuilder builder("AddAndTuple"); |
| 228 | auto p0 = xla::Parameter(&builder, 0, |
| 229 | xla::ShapeUtil::MakeShape(xla::F32, {2}), "P0"); |
| 230 | auto p1 = xla::Parameter(&builder, 1, |
| 231 | xla::ShapeUtil::MakeShape(xla::F32, {2}), "P1"); |
| 232 | auto sum = xla::Add(p0, p1); |
| 233 | xla::Tuple(&builder, {sum}); |
| 234 | return builder.Build().ValueOrDie(); |
| 235 | } |
| 236 | |
| 237 | xla::XlaComputation AddAndSubTuple() { |
| 238 | xla::XlaBuilder builder("AddAndSubTuple"); |