| 187 | } |
| 188 | |
| 189 | xla::XlaComputation SubAndScale() { |
| 190 | xla::XlaBuilder builder("SubAndScale"); |
| 191 | auto p0 = xla::Parameter(&builder, 0, |
| 192 | xla::ShapeUtil::MakeShape(xla::F32, {2}), "P0"); |
| 193 | auto p1 = xla::Parameter(&builder, 1, |
| 194 | xla::ShapeUtil::MakeShape(xla::F32, {2}), "P1"); |
| 195 | auto sum = xla::Sub(p0, p1); |
| 196 | auto c = xla::ConstantR0<float>(&builder, 11.0f); |
| 197 | xla::Mul(sum, c); |
| 198 | return builder.Build().ValueOrDie(); |
| 199 | } |
| 200 | |
| 201 | xla::XlaComputation Dot() { |
| 202 | xla::XlaBuilder builder("Dot"); |