| 175 | } |
| 176 | |
| 177 | xla::XlaComputation AddAndScale() { |
| 178 | xla::XlaBuilder builder("AddAndScale"); |
| 179 | auto p0 = xla::Parameter(&builder, 0, |
| 180 | xla::ShapeUtil::MakeShape(xla::F32, {2}), "P0"); |
| 181 | auto p1 = xla::Parameter(&builder, 1, |
| 182 | xla::ShapeUtil::MakeShape(xla::F32, {2}), "P1"); |
| 183 | auto sum = xla::Add(p0, p1); |
| 184 | auto c = xla::ConstantR0<float>(&builder, 3.0f); |
| 185 | xla::Mul(sum, c); |
| 186 | return builder.Build().ValueOrDie(); |
| 187 | } |
| 188 | |
| 189 | xla::XlaComputation SubAndScale() { |
| 190 | xla::XlaBuilder builder("SubAndScale"); |