| 471 | } |
| 472 | |
| 473 | FunctionDef XPlusOneXTimesY() { |
| 474 | const Tensor kOne = test::AsScalar<int64>(1); |
| 475 | return FDH::Define( |
| 476 | // Name |
| 477 | "XPlusOneXTimesY", |
| 478 | // Args |
| 479 | {"x: T", "y: T"}, |
| 480 | // Return values |
| 481 | {"s: T", "t: T"}, |
| 482 | // Attr def |
| 483 | {"T: {float, double, int32, int64}"}, |
| 484 | // Nodes |
| 485 | {{{"one"}, "Const", {}, {{"value", kOne}, {"dtype", DT_INT64}}}, |
| 486 | {{"increment"}, "Cast", {"one"}, {{"SrcT", DT_INT64}, {"DstT", "$T"}}}, |
| 487 | {{"s"}, "Add", {"x", "increment"}, {{"T", "$T"}}}, |
| 488 | {{"t"}, "Mul", {"x", "y"}, {{"T", "$T"}}}}); |
| 489 | } |
| 490 | |
| 491 | FunctionDef XYXLessThanOrEqualToN(int64 N) { |
| 492 | const Tensor kN = test::AsScalar<int64>(N); |